wafer-cli 0.2.14__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (41) hide show
  1. wafer/GUIDE.md +118 -0
  2. wafer/__init__.py +3 -0
  3. wafer/analytics.py +306 -0
  4. wafer/api_client.py +195 -0
  5. wafer/auth.py +432 -0
  6. wafer/autotuner.py +1080 -0
  7. wafer/billing.py +233 -0
  8. wafer/cli.py +7289 -0
  9. wafer/config.py +105 -0
  10. wafer/corpus.py +366 -0
  11. wafer/evaluate.py +4593 -0
  12. wafer/global_config.py +350 -0
  13. wafer/gpu_run.py +307 -0
  14. wafer/inference.py +148 -0
  15. wafer/kernel_scope.py +552 -0
  16. wafer/ncu_analyze.py +651 -0
  17. wafer/nsys_analyze.py +1042 -0
  18. wafer/nsys_profile.py +510 -0
  19. wafer/output.py +248 -0
  20. wafer/problems.py +357 -0
  21. wafer/rocprof_compute.py +490 -0
  22. wafer/rocprof_sdk.py +274 -0
  23. wafer/rocprof_systems.py +520 -0
  24. wafer/skills/wafer-guide/SKILL.md +129 -0
  25. wafer/ssh_keys.py +261 -0
  26. wafer/target_lock.py +270 -0
  27. wafer/targets.py +842 -0
  28. wafer/targets_ops.py +717 -0
  29. wafer/templates/__init__.py +0 -0
  30. wafer/templates/ask_docs.py +61 -0
  31. wafer/templates/optimize_kernel.py +71 -0
  32. wafer/templates/optimize_kernelbench.py +137 -0
  33. wafer/templates/trace_analyze.py +74 -0
  34. wafer/tracelens.py +218 -0
  35. wafer/wevin_cli.py +577 -0
  36. wafer/workspaces.py +852 -0
  37. wafer_cli-0.2.14.dist-info/METADATA +16 -0
  38. wafer_cli-0.2.14.dist-info/RECORD +41 -0
  39. wafer_cli-0.2.14.dist-info/WHEEL +5 -0
  40. wafer_cli-0.2.14.dist-info/entry_points.txt +2 -0
  41. wafer_cli-0.2.14.dist-info/top_level.txt +1 -0
wafer_cli-0.2.14.dist-info/METADATA ADDED
@@ -0,0 +1,16 @@
1
+ Metadata-Version: 2.4
2
+ Name: wafer-cli
3
+ Version: 0.2.14
4
+ Summary: CLI tool for running commands on remote GPUs and GPU kernel optimization agent
5
+ Requires-Python: >=3.11
6
+ Requires-Dist: typer>=0.12.0
7
+ Requires-Dist: trio>=0.24.0
8
+ Requires-Dist: trio-asyncio>=0.15.0
9
+ Requires-Dist: wafer-core>=0.1.0
10
+ Requires-Dist: perfetto>=0.16.0
11
+ Requires-Dist: posthog>=3.0.0
12
+ Provides-Extra: dev
13
+ Requires-Dist: pytest>=8.0.0; extra == "dev"
14
+ Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
15
+ Requires-Dist: diff-cover>=8.0.0; extra == "dev"
16
+ Requires-Dist: ruff>=0.4.0; extra == "dev"
wafer_cli-0.2.14.dist-info/RECORD ADDED
@@ -0,0 +1,41 @@
1
+ wafer/GUIDE.md,sha256=G6P4aFZslEXiHmVjtTB3_OIpGK5d1tSiqxtawASVUZg,3588
2
+ wafer/__init__.py,sha256=kBM_ONCpU6UUMBOH8Tmg4A88sNFnbaD59o61cJs-uYM,90
3
+ wafer/analytics.py,sha256=qLY6Z16usVHFD8TCv7XBuz7l47vXVdXk-qhOzA-hW_8,8179
4
+ wafer/api_client.py,sha256=i_Az2b2llC3DSW8yOL-BKqa7LSKuxOr8hSN40s-oQXY,6313
5
+ wafer/auth.py,sha256=nneKUjGwb5ggJEHRdF_GlFkT1ZozHP4kGyuXjhZjtgM,13677
6
+ wafer/autotuner.py,sha256=41WYP41pTDvMijv2h42vm89bcHtDMJXObDlWmn6xpFU,44416
7
+ wafer/billing.py,sha256=jbLB2lI4_9f2KD8uEFDi_ixLlowe5hasC0TIZJyIXRg,7163
8
+ wafer/cli.py,sha256=lBBTQCcmKREqZDOQh27qSq8i6NedjHW5oh1JiuT9aho,254241
9
+ wafer/config.py,sha256=h5Eo9_yfWqWGoPNdVQikI9GoZVUeysunSYiixf1mKcw,3411
10
+ wafer/corpus.py,sha256=x5aFhCsTSAtgzFG9AMFpqq92Ej63mXofL-vvvpjj1sM,12913
11
+ wafer/evaluate.py,sha256=bLTfL7jAGQlfqLL39hSGSB7bnBp5THTCY7nl6giVMkQ,176005
12
+ wafer/global_config.py,sha256=fhaR_RU3ufMksDmOohH1OLeQ0JT0SDW1hEip_zaP75k,11345
13
+ wafer/gpu_run.py,sha256=TwqXy72T7f2I7e6n5WWod3xgxCPnDhU0BgLsB4CUoQY,9716
14
+ wafer/inference.py,sha256=tZCO5i05FKY27ewis3CSBHFBeFbXY3xwj0DSjdoMY9s,4314
15
+ wafer/kernel_scope.py,sha256=YtnxknAChkJoeU_vIdxiqWsAITGBeabp9OGIK-X32i0,20796
16
+ wafer/ncu_analyze.py,sha256=rAWzKQRZEY6E_CL3gAWUaW3uZ4kvQVZskVCPDpsFJuE,24633
17
+ wafer/nsys_analyze.py,sha256=AhNcjPaapB0QCbqiHRXvyy-ccjevvVwEyxes84D28JU,36124
18
+ wafer/nsys_profile.py,sha256=QFBl8pkr8r4uRNdNUO9gY-obj9slqpOgVYFZ_sXu6Nw,15478
19
+ wafer/output.py,sha256=8jw5ifvIMK8ldyBMGW4NhrKvJPl66TV2Y2fJ5Tlhh1I,8293
20
+ wafer/problems.py,sha256=ce2sy10A1nnNUG3VGsseTS8jL7LZsku4dE8zVf9JHQ4,11296
21
+ wafer/rocprof_compute.py,sha256=n_yOGZaFbOXna_ghhmYWXeyUoSabgH4KkjlYq38DlHo,19888
22
+ wafer/rocprof_sdk.py,sha256=0Q7Ye6dUfa1anFZbqKc21rItgqva8V8VIZoSB7wqbmA,10085
23
+ wafer/rocprof_systems.py,sha256=4IWbMcbYk1x_8iS7P3FC_u5sgH6EXADCtR2lV9id80M,18629
24
+ wafer/ssh_keys.py,sha256=9kSdhV_dg9T6pQu2JmNQptarkkwGtN9rLyRkI1bW4i4,8094
25
+ wafer/target_lock.py,sha256=SDKhNzv2N7gsphGflcNni9FE5YYuAMuEthngAJEo4Gs,7809
26
+ wafer/targets.py,sha256=9r-iRWoKSH5cQl1LcamaX-T7cNVOg99ngIm_hlRk-qU,26922
27
+ wafer/targets_ops.py,sha256=jN1oIBx0mutxRNE9xpIc7SaBxPkVmOyus2eqn0kEKNI,21475
28
+ wafer/tracelens.py,sha256=g9ZIeFyNojZn4uTd3skPqIrRiL7aMJOz_-GOd3aiyy4,7998
29
+ wafer/wevin_cli.py,sha256=vF3GNH-qWXO4hAlXaDg98VZpS4uFexVUp94BHsJjjMU,22179
30
+ wafer/workspaces.py,sha256=XZvN-13oq40fkpoJTB2UWTG9KkD-eO47ptXK0FY6360,30083
31
+ wafer/skills/wafer-guide/SKILL.md,sha256=KWetJw2TVTbz11_nzqazqOJWWRlbHRFShs4sOoreiWo,3255
32
+ wafer/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
33
+ wafer/templates/ask_docs.py,sha256=Lxs-faz9v5m4Qa4NjF2X_lE8KwM9ES9MNJkxo7ep56o,2256
34
+ wafer/templates/optimize_kernel.py,sha256=u6AL7Q3uttqlnBLzcoFdsiPq5lV2TV3bgqwCYYlK9gk,2357
35
+ wafer/templates/optimize_kernelbench.py,sha256=aoOA13zWEl89r6QW03xF9NKxQ7j4mWe9rwua6-mlr4Y,4780
36
+ wafer/templates/trace_analyze.py,sha256=XE1VqzVkIUsZbXF8EzQdDYgg-AZEYAOFpr6B_vnRELc,2880
37
+ wafer_cli-0.2.14.dist-info/METADATA,sha256=Ul8te_86H-H1ZWNrK7P61EU_wlBIpS6qPIkygJV1zjg,560
38
+ wafer_cli-0.2.14.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
39
+ wafer_cli-0.2.14.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
40
+ wafer_cli-0.2.14.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
41
+ wafer_cli-0.2.14.dist-info/RECORD,,
wafer_cli-0.2.14.dist-info/WHEEL ADDED
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: setuptools (80.10.2)
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
5
+
wafer_cli-0.2.14.dist-info/entry_points.txt ADDED
@@ -0,0 +1,2 @@
1
+ [console_scripts]
2
+ wafer = wafer.cli:main
wafer_cli-0.2.14.dist-info/top_level.txt ADDED
@@ -0,0 +1 @@
1
+ wafer