shark-turbine

View on PyPIReverse Dependencies (0)

2.4.1 shark_turbine-2.4.1-py3-none-any.whl

Wheel Details

Project: shark-turbine
Version: 2.4.1
Filename: shark_turbine-2.4.1-py3-none-any.whl
Download: [link]
Size: 223086
MD5: 21e33c229290f0b0bf1996fd3d83da63
SHA256: 7ee28f64892493b30dad1ffbf1ae17fedcf39c1b83afa3f3100325d90746ddca
Uploaded: 2024-09-03 22:46:20 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: shark-turbine
Version: 2.4.1
Summary: SHARK Turbine Machine Learning Deployment Tools
Author: SHARK Authors
Author-Email: stella[at]nod.ai
Home-Page: https://github.com/nod-ai/SHARK-Turbine
License: Apache-2.0
Classifier: Development Status :: 5 - Production/Stable
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Requires-Dist: numpy (>=1.26.3)
Requires-Dist: iree-compiler
Requires-Dist: iree-runtime
Requires-Dist: torch (>=2.3.0)
Requires-Dist: Jinja2 (>=3.1.3)
Requires-Dist: pytest (>=8.0.0); extra == "testing"
Requires-Dist: pytest-xdist (>=3.5.0); extra == "testing"
Requires-Dist: parameterized (>=0.9.0); extra == "testing"
Provides-Extra: testing
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4639 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (74.1.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
iree/turbine/__init__.py sha256=UQdKc1VCmcp1zMpbrFqFHcSSr1E5MjPpwNA6S2zD6fw 831
shark_turbine/aot/__init__.py sha256=HS96VEWDRqgxNGsV8XFBlexBGbBnFe_0zjYrVOv6zo8 515
shark_turbine/aot/compiled_module.py sha256=uYIrK3lb-Jr4aiOrd5ljnMfkLc-qcFAgfXFrHoPwK4Y 23347
shark_turbine/aot/decompositions.py sha256=ouYk2ccigcWNtxo-oy3jSgJNBK1wG42fMJSQxZrqVo8 2815
shark_turbine/aot/exporter.py sha256=v56hSN6s744fhBtl8ys0Gpa26btjYU5187n7m-12rvE 10752
shark_turbine/aot/fx_programs.py sha256=jrRZqZDbCot8bu3L5JDn21bo8kf2ZGOWLr9RyjNp3AI 12219
shark_turbine/aot/params.py sha256=7GbQ70xoiiC1-8M606Ytllpp3yIZHOAPL0pClTpOnmM 10288
shark_turbine/aot/tensor_traits.py sha256=u_IuLi_SAJsh7BxCwiWueYbA0gxNIbdRRZCIFASnfP0 1023
shark_turbine/aot/builtins/__init__.py sha256=rDNDLuRqs_HmZE6QF1SJL1eszt-IhHivcraKqI7Pt9Q 906
shark_turbine/aot/builtins/globals.py sha256=v-SevLOve3je1IPUTDeHKreJm6-A9XQoCCmRyoqAPf0 7100
shark_turbine/aot/builtins/jittable.py sha256=AU6bSBtW1sXc8e0WTbIuqpBe6_HqTxbSfLB5ZhMnN88 17423
shark_turbine/aot/passes/__init__.py sha256=mkiANoenJUYzb2NsRgQV4sDfVPi-0b9VWotNDJ6ayHI 262
shark_turbine/aot/passes/functorch.py sha256=SwzWVkWLDq7ud_y-kODNEB1wPk_EaGsaHCdCqYQyr1w 3537
shark_turbine/aot/support/ir_utils.py sha256=Lqy2PV2viWfD5ZMT-d-ZlbgqrwODAPAhu2VpuhuBTMA 16691
shark_turbine/aot/support/procedural/__init__.py sha256=AAKJuYXBVmBbq-cClOOlzh3kbRiGW-Bkbwj982n8BiA 646
shark_turbine/aot/support/procedural/base.py sha256=2F5wdROfa75OQ96Y8by86FzoY5trK3C-fgrplWhC_ZQ 7507
shark_turbine/aot/support/procedural/exported_program.py sha256=A8eNKl9IYtBebQmX_Q8CrKo9iG9SDMT2WbNjpZE71Kg 12631
shark_turbine/aot/support/procedural/globals.py sha256=rmQoFOEsjDsLZZ7PtjSBu69E6rumHrW-Vpme34DmcqI 9222
shark_turbine/aot/support/procedural/iree_emitter.py sha256=A0AU6KuE8r1A9HTfq6KHZkluO-ZAY3Dr8rwfB7GCGV4 12373
shark_turbine/aot/support/procedural/primitives.py sha256=O2xh2FdDg23cAbzUXMuH8uG1CzESwhH7sG85CXtsHNo 11744
shark_turbine/aot/support/procedural/tracer.py sha256=Eq6RVrl0wUVu1qUXxO5Aal6F-BUUuwspVWDRV3HeKcI 7211
shark_turbine/dynamo/__init__.py sha256=fsprbtYTtyX2jxbs8O8VZqutbrDJG87T1YkAGhwXisk 287
shark_turbine/dynamo/decompositions.py sha256=dN2CFWML9moV_zgWWFEpXuo3IsYPjvD4tuhOgKqrvnw 4022
shark_turbine/dynamo/executor.py sha256=O2cFnLIltcaoD4cvlEUd-ngchNgvJrwLbQjDtTcuJeQ 8254
shark_turbine/dynamo/passes.py sha256=jFZdcVBL-fOaIJfjp2J66_iD1Hz2j5HRaM9JEo09WgE 814
shark_turbine/dynamo/tensor.py sha256=2w7QuhIG_55-wa-Kc4sfcLCk-9PpdQzjDygtNr223H8 21298
shark_turbine/dynamo/type_conversion.py sha256=jKzPP1Z28FrUYRDEiqbeyQNj7r4Nbyc166CVF2lI5EM 7737
shark_turbine/dynamo/backends/cpu.py sha256=i_1bYvn8tEisN32s9vJ2lDClK8h1bgtPEc59nNu5la8 2816
shark_turbine/importers/ir.py sha256=ufPQWLqroZ6Ul05k1RXLQHvstzAUD46IxABrIKtHaJM 857
shark_turbine/importers/utils.py sha256=9K8S7Wu7R3kqgiU8r_2hBET1VKGgQgTyv_0svK7Z_bw 1192
shark_turbine/kernel/__init__.py sha256=n-87s3IbAj0UNlIGshgz-4UxfI1t3RL-9Q6ncnl78nk 537
shark_turbine/kernel/_support/context.py sha256=7lr1iRE1YEJyYP8bBTHh5dyN_itoOpKUd7C0VK0LaBo 1484
shark_turbine/kernel/_support/dtype.py sha256=odq8N5BfKI_u-gQAUP82VxTXqMIRaL8E_5LeJZKXdtc 1327
shark_turbine/kernel/_support/indexing.py sha256=NUbne4D-KR33jo5XVsijXhQ9euojqxOuqcdAHR92IZY 13311
shark_turbine/kernel/_support/regions.py sha256=TvQk6v2KFPsxcVrXZz-Z-bDr2tYAsktSmaAJVpLgLHc 5147
shark_turbine/kernel/_support/shaped_type.py sha256=crompGTpkHrjn_7uZwfEMiOwghKR_XxaPl0Dmxb5LOk 3549
shark_turbine/kernel/_support/tracing.py sha256=7TCGrOTVTKKEP7X6yMWJwDtXfliNHvyky9tnayADHjk 16710
shark_turbine/kernel/compiler/base.py sha256=8D8LnJrBROm6JILMsUIbNBEXs0pw_Pr1T1deOwQx3Ds 103
shark_turbine/kernel/compiler/builder.py sha256=dbhW728w3NEIyBSyzGsh-D0yLnI8IdzjUEcBR0YHbOo 11457
shark_turbine/kernel/compiler/dispatch_codegen.py sha256=nsh7uuPKdpKrZKeHMkoCAM5uqFLLTKi8VM7oyoKVos0 8766
shark_turbine/kernel/compiler/host_codegen.py sha256=0oqcY5KOouJPqCDXIvaChTSZeERyzJ2auHX9GmGi3VU 1796
shark_turbine/kernel/compiler/ir.py sha256=h0RreQ15ajco2ii2zInLu7c-enqGZi8WZqB-tJGl96g 868
shark_turbine/kernel/compiler/kernel_codegen.py sha256=UOhRks-ZRxuny_9YBxjyTcXFijMI1szkWT7PU6lSt38 11169
shark_turbine/kernel/compiler/op_matchers.py sha256=7zO7EPmaF5KgcpL5GP7SawfLTxa5QSwtREf6IoSOcZw 1770
shark_turbine/kernel/compiler/utils.py sha256=C1ulG-QJNqPyDN4ZDGZuT46NUlxWNLzATLiJ4bSP5MU 726
shark_turbine/kernel/compiler/vector_codegen.py sha256=u21q_SBA6XZUMnzuXQP0RVWXHuos5T2NAoB6GE8z4yQ 36317
shark_turbine/kernel/gen/__init__.py sha256=jWP3DXM5Md0RdF-78lbhDZXJemrz-UPZwt5nYAAOz10 94
shark_turbine/kernel/gen/kernel.py sha256=fXYMyIzkoqZw5fFC5ax4dXmZ8MwyYXLOaMjhS-ejLRQ 5342
shark_turbine/kernel/gen/thread.py sha256=EofUjcnFYlHBA0sripukJ_ldnUqG6SID52cjdKFvg9I 4886
shark_turbine/kernel/lang/__init__.py sha256=2Pi_iP5kXNGCA-xip9bf2vofKO7qqb0LAW_FT-1I954 420
shark_turbine/kernel/lang/global_symbols.py sha256=tY8UcUkmb8vgaAryYmoSIG2GYlQbUI9W0831eCMFKlw 574
shark_turbine/kernel/lang/grid.py sha256=JaWasWVPiAoSbvuir0iXyioksKeLl-4n1d5-PFRIqFU 1454
shark_turbine/kernel/lang/kernel_buffer.py sha256=MeOku-NdUeRANHGrx9C6P1NoOj6eLYt8DCreihzmTyY 4930
shark_turbine/kernel/lang/prims.py sha256=EV_rgSXSixmgFqwWUXrC7F7IPvIKGQy_qUQr9OSKebs 1031
shark_turbine/kernel/lang/types.py sha256=qCEr7fp-67D76x7I4CHx-Bqn59nVz4Yx94t0ir4CMcs 1331
shark_turbine/kernel/lang/wave_types.py sha256=-7aWkX86zh_Df7r5znHNKI_X7nNwbVmcCKHko-9eEqY 7500
shark_turbine/kernel/ops/__init__.py sha256=2TPflRWKzKhH1VQnNjBtuL6Vs0SLKbMIwRUaA-foyf4 149
shark_turbine/kernel/ops/base.py sha256=9RnxNDhertSjYQSzqsf0V89gnxzEufoj_SmmXwH6L8w 1313
shark_turbine/kernel/ops/control_flow.py sha256=fHcTKIWB_0j75XM1Y_sFnB2RcmQ5pM8Zbrx2ovMZxDE 638
shark_turbine/kernel/ops/core.py sha256=N-gClns5wt9SKoYI_OAYVV9eH7sVa8vfrtBBHs3tgmU 581
shark_turbine/kernel/ops/math.py sha256=O7n8wXQHYvU9x00WN0J1xJeSdKd481JBhsBxkppwxQc 330
shark_turbine/kernel/ops/memory.py sha256=ZtyNLdWsDC_xkYyxqzs5bGp1wnQjCOGT0J7QfFRh4VI 577
shark_turbine/kernel/ops/reduction.py sha256=AfJHtJM6Iwy3dO6chdi4J1A04PZ6VkEwuSof8sgXY0s 487
shark_turbine/kernel/ops/shape_manipulation.py sha256=IswoyHQ1Zt6LHZrg4i0yI9CfW4v_0jxj2yrS4xmP5g0 555
shark_turbine/kernel/ops/wave_ops.py sha256=KHFb5kX-kN1eY32ExjnndBHf6bYKOq9QW9f_RvDF7bg 31614
shark_turbine/kernel/wave/__init__.py sha256=hOUYALCNbaSdK4gF3EEqT8cIyF6xYL56LYfzoSmxo9o 107
shark_turbine/kernel/wave/barriers.py sha256=6JkpzfyHF-NUMgrCzYkS0PUa1VqMfFBx2B8xZBBO0dE 1928
shark_turbine/kernel/wave/codegen.py sha256=bZZqbf4FsapOUE6ykoQ4M-Socy7BGEJZ24q9VpRk5B4 31210
shark_turbine/kernel/wave/constraints.py sha256=fChBx1i-WgD7Y4E6neDh3_C9Nzmag6KYzI7Ho3SzSFE 10201
shark_turbine/kernel/wave/decompose_reduce_ops.py sha256=IlCqZVhTO5vxU2h_ia2gN8QzYe1Y56bJPSYS8GOVt6U 4129
shark_turbine/kernel/wave/expansion.py sha256=G-VPZLzAQqQyfVVHAZDrXUmCGZJqOt1ASaVBKIZ0XsQ 20898
shark_turbine/kernel/wave/hoisting.py sha256=Kcx5dPnYOSrMWuU_Q1PZ_MX8o-oNMtSIT3M1Xku0wzQ 1121
shark_turbine/kernel/wave/index_sequence_analysis.py sha256=fXomuZEw8jxx57GrmkHyxLBOa5xxDXED4bdz-XdvtmY 3295
shark_turbine/kernel/wave/indexing.py sha256=9cMbfl8Sj4gcSsnEcElvyYGbGjYm4CZgqFe9OC5adAc 1091
shark_turbine/kernel/wave/iree_utils.py sha256=P12NhHRaaUbFuUTSzw6OBr1oM3YMj1a5Lpc6-dNK9To 1926
shark_turbine/kernel/wave/minimize_global_loads.py sha256=xtLv5SctNZJkaboEJJYrjWl2ihd10emf5zJ-ovy_E-8 8604
shark_turbine/kernel/wave/promotion.py sha256=D5zkeRNn7pSHpNDKOK-bhTDwUjRSGauBxC-h2JHNsS4 2599
shark_turbine/kernel/wave/register_analysis.py sha256=roGkR_ILfkYKxCXTB_t6xtzOxbpKGt_sX5uniNdSc2s 2134
shark_turbine/kernel/wave/utils.py sha256=qPYCndzUYANtYdah17m-fXBsGoD5gGe5mN_gaEQfFrQ 10234
shark_turbine/kernel/wave/visualization.py sha256=QY4WYExpqefWrIGPiKJYi14nCrjjeh_CLNHhIGY_fus 752
shark_turbine/kernel/wave/wave.py sha256=gW6aAuWZAf4l1tpzHYFHtE292vx4lpnTOxi0tiM4QOo 10338
shark_turbine/kernel/wave/wave_sim.py sha256=gxJViFXNJDNDDCsb1ORm2ZzX01wc30IzaCBSnPExC64 8981
shark_turbine/ops/__init__.py sha256=S-WySt-YqNaV8xGYtVqzWDdDM4Bf6Mq7uRbt1Iea8CI 313
shark_turbine/ops/_jinja_test_ops.py sha256=fSHwzR8j1HCx_sJ6T5PIzsaUX_CeVhMmWEViUVf45iw 1573
shark_turbine/ops/_str_format_test_ops.py sha256=-7CQrQhnhC4uWP1ZqP9bdcZwwaV7mBU1rzsJYqGbmxw 2311
shark_turbine/ops/iree.py sha256=j2EwpNKC6ykW7SUveTfdam1zswPaJJGL0kXvXFkwNMo 4330
shark_turbine/ops/templates/test_add_jinja.mlir sha256=HQ4GP9nWf-XQ4IPA9tgabFA52wZZbveC4w63_Ynbi-s 347
shark_turbine/ops/templates/test_add_strformat.mlir sha256=Ce0NPsKwWVi-MI-8L1-tU9bEmDZ1QD_pUVcmshsTAOg 349
shark_turbine/ops/templates/test_syntax_error.mlir sha256=hWw5j1M684WXWKdVoF5UtYYFmufof2TdZ7o6PjGRqDM 23
shark_turbine/runtime/__init__.py sha256=uBTaMM-yjqGS8yvmhw9m5qjWPhstYqvr4kW9eynDxEc 294
shark_turbine/runtime/device.py sha256=pHT7EYjmyUEQstQzQEIEp-r1k8J4ZnnKYiO5YrIT0lM 19020
shark_turbine/runtime/launch.py sha256=2GPKUKcZMZYazaj8vXhQlXISzg05hN3P15QN1SiPkPY 9459
shark_turbine/runtime/tracing.py sha256=gzTdCBXaC1gZj7ibFgwFVIk02-RSebU8DgIfblNnkx4 3453
shark_turbine/runtime/op_reg/__init__.py sha256=ydBd9F07r28wLd0BFTCXxWLFbW2RC37EKc7iIE1PEZ4 275
shark_turbine/runtime/op_reg/base.py sha256=EUWgVY9kt-OEcnhbLjHJd_wgQi6ra5QJaC5QPMRxw5w 32847
shark_turbine/runtime/op_reg/compiler.py sha256=4e63aGDufZkBCzOVqWJaISreocO8yrmLntSPtadxXSU 5344
shark_turbine/runtime/op_reg/eager.py sha256=0K8ohFvSkPUzrtnWQLqrMLXGbJ54bEtZLWqY-Mtk_ac 8341
shark_turbine/runtime/op_reg/impl_helper.py sha256=heFsG8pWbal2EqrFGMP6sXtP-ro1VrHzpgOyOmJ9ua4 5339
shark_turbine/support/__init__.py sha256=4XzJ8ZeS0fbKqFsLXnyKOJ622sfrZZZiTWIRMFCyEw4 339
shark_turbine/support/conversions.py sha256=xZvirjOB8V7cgW82zPeK5sMv0EsBJLIvug8jMiJ2WfM 4833
shark_turbine/support/debugging.py sha256=Io3Z0K-KtA014PX1Hp8WJN_Acj8xAOX9OeIQfP6W2pM 3182
shark_turbine/support/exceptions.py sha256=EhMtpUEWT4wXo1ANF688oQP5H4ry2it9fjnEhJ9Nd24 1341
shark_turbine/support/ir_imports.py sha256=0X3XsMl3Uqk-h6fiYtNx07J2IN4_JnMRqAeYHis87nE 1256
shark_turbine/support/logging.py sha256=2HkyMX1kF41yukyU7trjxaFzHmcoJ4CvWAJpNirzA6c 1255
shark_turbine/transforms/builder.py sha256=YDDfIZF28REmEUEyLRBRF5Y1oNyu9QKM59_snzgVI8w 2365
shark_turbine/transforms/merger.py sha256=yIEIaDZY1JaajN8TQoX8Vc7LWGVLWeUC7zl_fSYbHYY 8382
shark_turbine/transforms/rewriter.py sha256=UfTg26XJONA7dIipzyV6TMxq4Wgr2I7TxhqrWMwyktY 10997
shark_turbine/transforms/general/add_metadata.py sha256=czu9qmRiX9RNPkQNWF77U9AWXRAhdw9cRzRPky-oQM8 1909
shark_turbine/transforms/general/custom_op_expansion.py sha256=_Ei2V9hY3HoU0YsOCySzWxEZkmeurNJqVlj9GvahYSo 14776
shark_turbine/transforms/general/rename_parameters.py sha256=bPmDqRuCMJJFN5SjP8-b0ajcfDpIx5zpSKQZ_nB-jbg 4995
shark_turbine/transforms/quantization/mm_group_quant.py sha256=Jrhak4ip_vtIGZpgan9vIsg1Zd8yD1Fl9mPFXIIlK-8 8092
shark_turbine-2.4.1.dist-info/LICENSE sha256=nkXoVr7czun2clQILKEYUdlU3i_tdEjEvtGa2aq5mpE 12262
shark_turbine-2.4.1.dist-info/METADATA sha256=aDvYbl96yU4DmjRaW3ynBFrbAVufuB2Q3j-ia4DHClE 5431
shark_turbine-2.4.1.dist-info/WHEEL sha256=uCRv0ZEik_232NlR4YDw4Pv3Ajt5bKvMH13NUU7hFuI 91
shark_turbine-2.4.1.dist-info/entry_points.txt sha256=-bxFqt7ywvSixYOWlz1sZ8MXChj-5joPPR_gZfT3Kn4 80
shark_turbine-2.4.1.dist-info/top_level.txt sha256=HC8-angiduPf60mk4TnoHh0l6DedI5geRT925eyi6mc 19
shark_turbine-2.4.1.dist-info/RECORD

top_level.txt

iree
shark_turbine

entry_points.txt

turbine_cpu = shark_turbine.dynamo.backends.cpu:backend