quack-kernels 0.2.3__tar.gz → 0.2.4__tar.gz

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 (60) hide show
  1. {quack_kernels-0.2.3/quack_kernels.egg-info → quack_kernels-0.2.4}/PKG-INFO +3 -3
  2. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/pyproject.toml +2 -2
  3. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/__init__.py +1 -1
  4. {quack_kernels-0.2.3 → quack_kernels-0.2.4/quack_kernels.egg-info}/PKG-INFO +3 -3
  5. quack_kernels-0.2.4/quack_kernels.egg-info/requires.txt +8 -0
  6. quack_kernels-0.2.3/quack_kernels.egg-info/requires.txt +0 -8
  7. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/LICENSE +0 -0
  8. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/README.md +0 -0
  9. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/activation.py +0 -0
  10. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/autotuner.py +0 -0
  11. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/broadcast_utils.py +0 -0
  12. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/compile_utils.py +0 -0
  13. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/copy_utils.py +0 -0
  14. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/cross_entropy.py +0 -0
  15. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/cute_dsl_utils.py +0 -0
  16. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/fast_math.py +0 -0
  17. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm.py +0 -0
  18. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_act.py +0 -0
  19. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_config.py +0 -0
  20. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_dact.py +0 -0
  21. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_default_epi.py +0 -0
  22. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_interface.py +0 -0
  23. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_sm100.py +0 -0
  24. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_sm90.py +0 -0
  25. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_symmetric.py +0 -0
  26. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/gemm_wrapper_utils.py +0 -0
  27. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/layout_utils.py +0 -0
  28. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/linear.py +0 -0
  29. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/linear_cross_entropy.py +0 -0
  30. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/mlp.py +0 -0
  31. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/pipeline.py +0 -0
  32. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/reduce.py +0 -0
  33. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/reduction_base.py +0 -0
  34. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/rmsnorm.py +0 -0
  35. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/sm100_utils.py +0 -0
  36. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/sm90_utils.py +0 -0
  37. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/softmax.py +0 -0
  38. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/sort/bitonic_sort.py +0 -0
  39. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/sort/generate_sorting_networks.py +0 -0
  40. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/sort/sorting_networks.py +0 -0
  41. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/sort/utils.py +0 -0
  42. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/tensormap_manager.py +0 -0
  43. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/tile_scheduler.py +0 -0
  44. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/topk.py +0 -0
  45. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/utils.py +0 -0
  46. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/varlen_utils.py +0 -0
  47. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack_kernels.egg-info/SOURCES.txt +0 -0
  48. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack_kernels.egg-info/dependency_links.txt +0 -0
  49. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack_kernels.egg-info/top_level.txt +0 -0
  50. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/setup.cfg +0 -0
  51. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_cross_entropy.py +0 -0
  52. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_layernorm.py +0 -0
  53. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_linear.py +0 -0
  54. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_linear_cross_entropy.py +0 -0
  55. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_linear_varlen_k.py +0 -0
  56. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_linear_varlen_m.py +0 -0
  57. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_rmsnorm.py +0 -0
  58. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_softmax.py +0 -0
  59. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_symmetric_gemm.py +0 -0
  60. {quack_kernels-0.2.3 → quack_kernels-0.2.4}/tests/test_topk.py +0 -0
{quack_kernels-0.2.3/quack_kernels.egg-info → quack_kernels-0.2.4}/PKG-INFO RENAMED
@@ -1,11 +1,11 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: quack-kernels
3
- Version: 0.2.3
3
+ Version: 0.2.4
4
4
  Requires-Python: >=3.10
5
5
  License-File: LICENSE
6
- Requires-Dist: nvidia-cutlass-dsl==4.3.3
6
+ Requires-Dist: nvidia-cutlass-dsl<4.4.0,>=4.3.4
7
7
  Requires-Dist: torch
8
- Requires-Dist: apache-tvm-ffi<0.2,>=0.1.5
8
+ Requires-Dist: apache-tvm-ffi<0.2,>=0.1.6
9
9
  Requires-Dist: torch-c-dlpack-ext
10
10
  Provides-Extra: dev
11
11
  Requires-Dist: pre-commit; extra == "dev"
{quack_kernels-0.2.3 → quack_kernels-0.2.4}/pyproject.toml RENAMED
@@ -7,9 +7,9 @@ name = "quack-kernels"
7
7
  dynamic = ["version"]
8
8
  requires-python = ">=3.10"
9
9
  dependencies = [
10
- "nvidia-cutlass-dsl==4.3.3",
10
+ "nvidia-cutlass-dsl>=4.3.4,<4.4.0",
11
11
  "torch",
12
- "apache-tvm-ffi>=0.1.5,<0.2",
12
+ "apache-tvm-ffi>=0.1.6,<0.2",
13
13
  "torch-c-dlpack-ext",
14
14
  ]
15
15
 
{quack_kernels-0.2.3 → quack_kernels-0.2.4}/quack/__init__.py RENAMED
@@ -1,4 +1,4 @@
1
- __version__ = "0.2.3"
1
+ __version__ = "0.2.4"
2
2
 
3
3
  from quack.rmsnorm import rmsnorm
4
4
  from quack.softmax import softmax
{quack_kernels-0.2.3 → quack_kernels-0.2.4/quack_kernels.egg-info}/PKG-INFO RENAMED
@@ -1,11 +1,11 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: quack-kernels
3
- Version: 0.2.3
3
+ Version: 0.2.4
4
4
  Requires-Python: >=3.10
5
5
  License-File: LICENSE
6
- Requires-Dist: nvidia-cutlass-dsl==4.3.3
6
+ Requires-Dist: nvidia-cutlass-dsl<4.4.0,>=4.3.4
7
7
  Requires-Dist: torch
8
- Requires-Dist: apache-tvm-ffi<0.2,>=0.1.5
8
+ Requires-Dist: apache-tvm-ffi<0.2,>=0.1.6
9
9
  Requires-Dist: torch-c-dlpack-ext
10
10
  Provides-Extra: dev
11
11
  Requires-Dist: pre-commit; extra == "dev"
quack_kernels-0.2.4/quack_kernels.egg-info/requires.txt ADDED
@@ -0,0 +1,8 @@
1
+ nvidia-cutlass-dsl<4.4.0,>=4.3.4
2
+ torch
3
+ apache-tvm-ffi<0.2,>=0.1.6
4
+ torch-c-dlpack-ext
5
+
6
+ [dev]
7
+ pre-commit
8
+ ruff
quack_kernels-0.2.3/quack_kernels.egg-info/requires.txt DELETED
@@ -1,8 +0,0 @@
1
- nvidia-cutlass-dsl==4.3.3
2
- torch
3
- apache-tvm-ffi<0.2,>=0.1.5
4
- torch-c-dlpack-ext
5
-
6
- [dev]
7
- pre-commit
8
- ruff
{quack_kernels-0.2.3 → quack_kernels-0.2.4}/LICENSE RENAMED
File without changes
{quack_kernels-0.2.3 → quack_kernels-0.2.4}/README.md RENAMED
File without changes
{quack_kernels-0.2.3 → quack_kernels-0.2.4}/setup.cfg RENAMED
File without changes