cuequivariance-ops-cu12 0.4.0__py3-none-manylinux_2_39_aarch64.whl → 0.5.0__py3-none-manylinux_2_39_aarch64.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.

Potentially problematic release.

This version of cuequivariance-ops-cu12 might be problematic. Click here for more details.

Files changed (19) hide show
  1. cuequivariance_ops/VERSION +1 -1
  2. cuequivariance_ops/__init__.py +3 -2
  3. cuequivariance_ops/cache_manager.py +130 -0
  4. cuequivariance_ops/equivariance/dtypes.hh +21 -0
  5. cuequivariance_ops/equivariance/indexed_linear.hh +36 -0
  6. cuequivariance_ops/equivariance/run_fmha.h +192 -0
  7. cuequivariance_ops/equivariance/run_fmha_cudafree.h +77 -0
  8. cuequivariance_ops/equivariance/tensor_product_uniform_1d_jit.hh +17 -35
  9. cuequivariance_ops/fused_layer_norm_triton.py +324 -0
  10. cuequivariance_ops/fused_sigmoid_gated_dual_gemm_backward_pregemm_kernel_wrapper.json +222844 -0
  11. cuequivariance_ops/fused_sigmoid_gated_dual_gemm_forward_kernel_wrapper.json +326932 -0
  12. cuequivariance_ops/gated_gemm_triton.py +340 -0
  13. cuequivariance_ops/lib/libcue_ops.so +0 -0
  14. cuequivariance_ops/tuning_decorator.py +328 -0
  15. {cuequivariance_ops_cu12-0.4.0.dist-info → cuequivariance_ops_cu12-0.5.0.dist-info}/METADATA +4 -1
  16. cuequivariance_ops_cu12-0.5.0.dist-info/RECORD +23 -0
  17. {cuequivariance_ops_cu12-0.4.0.dist-info → cuequivariance_ops_cu12-0.5.0.dist-info}/WHEEL +1 -1
  18. cuequivariance_ops_cu12-0.4.0.dist-info/RECORD +0 -13
  19. {cuequivariance_ops_cu12-0.4.0.dist-info → cuequivariance_ops_cu12-0.5.0.dist-info}/licenses/LICENSE +0 -0
cuequivariance_ops_cu12-0.5.0.dist-info/RECORD ADDED
@@ -0,0 +1,23 @@
1
+ cuequivariance_ops/VERSION,sha256=oK1QZAE5pST4ZZEVcUW_HUZ06pwGW_6iFVjw97BEMMg,6
2
+ cuequivariance_ops/__init__.py,sha256=wvvAMuXpOg5W4oE-AnHDWoHPzcamAWK_DiUXyg3hgW8,1332
3
+ cuequivariance_ops/_version.py,sha256=o9Flao_mTq2Y7TrrjnSCqEAgebmA0sGozsl15qVI13Y,730
4
+ cuequivariance_ops/cache_manager.py,sha256=up3F0lbVoQ0VtG7Zg0kmZo18rLdSVAjOL-kq6m2YrUo,4576
5
+ cuequivariance_ops/fused_layer_norm_triton.py,sha256=SyQf_eJvTKm3Foe8BI0sjWZdEtGp__LQ1qgImlHLc4c,11056
6
+ cuequivariance_ops/fused_sigmoid_gated_dual_gemm_backward_pregemm_kernel_wrapper.json,sha256=NoLX8fVB3g-7Vc1bTSg7DBggivh4SPxQncSxrQ954Vg,8360746
7
+ cuequivariance_ops/fused_sigmoid_gated_dual_gemm_forward_kernel_wrapper.json,sha256=pXPRSrKZPCjgo4fQffyKZ96E6CHDpE3hZ9SQJiWh3LI,11813895
8
+ cuequivariance_ops/gated_gemm_triton.py,sha256=PEJcgNVZUk8G5Z5ukD8Ksbe71kUWHUufjUINI5JGnV8,11405
9
+ cuequivariance_ops/tuning_decorator.py,sha256=7eogU9gp-vOKjjUl72ERaWHoL5zj9GSo4jh1GvnHKPg,12823
10
+ cuequivariance_ops/common/common.hpp,sha256=2zDyE5lGugQL43vmM4_ylmp-Tz8OBFnPRsdFra_1BdM,2787
11
+ cuequivariance_ops/common/nvtx.hpp,sha256=Wi6z9b-yFUNq6ShJjjcsdxQRqCygd4xGegGJrqUI9Wk,708
12
+ cuequivariance_ops/equivariance/dtypes.hh,sha256=w0BYWZ0LYklODXhp7PR6VYE__DE1Syj0Ur11aFaq9VM,466
13
+ cuequivariance_ops/equivariance/fused_tensor_product.cuh,sha256=bOXR5UWU9gNYRfdh6k28NEkV3CUU2ijmh6y7c0ND0J4,8283
14
+ cuequivariance_ops/equivariance/indexed_linear.hh,sha256=lNqJNafJdPyMAUp6iwWvu6RyassSXh7JqyqJ4bfjoxQ,1402
15
+ cuequivariance_ops/equivariance/run_fmha.h,sha256=7l62dTQJbX7BbHLB7MmVP1t26Cfpmcu3h6eY048Hof0,9505
16
+ cuequivariance_ops/equivariance/run_fmha_cudafree.h,sha256=bF2_nrvSfrqSVZ0eOPcq4CJ-NKqmJ2VgQv1cstvHBkU,2695
17
+ cuequivariance_ops/equivariance/segmented_transpose.cuh,sha256=gfSZhRBwSqwVAgFCCiGtI-NJ8yDy9tV_iCg1G2KpctY,1766
18
+ cuequivariance_ops/equivariance/tensor_product_uniform_1d_jit.hh,sha256=7PPybCWczS58XKA-iFLoCM7MDEomO4-enF6RCBj5G5M,1922
19
+ cuequivariance_ops/lib/libcue_ops.so,sha256=rSk0Km-M-Zu703bOMV-OF8mykMPOgFTyx73OW_r8ofM,112269112
20
+ cuequivariance_ops_cu12-0.5.0.dist-info/METADATA,sha256=vuyCmbrl0bzSccJAfbXAfW5WqUv_gKt9NEtGjENK7tk,20926
21
+ cuequivariance_ops_cu12-0.5.0.dist-info/WHEEL,sha256=RxM28Avh4PDgHOLX-AZLV1MP0dIb1yycxVPEx6_SFW0,116
22
+ cuequivariance_ops_cu12-0.5.0.dist-info/RECORD,,
23
+ cuequivariance_ops_cu12-0.5.0.dist-info/licenses/LICENSE,sha256=rvp0QV9FuOdxz_CGWTd9DgId4xh2BByyXfBBnb0ejZM,18279
{cuequivariance_ops_cu12-0.4.0.dist-info → cuequivariance_ops_cu12-0.5.0.dist-info}/WHEEL RENAMED
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: scikit-build-core 0.11.1
2
+ Generator: scikit-build-core 0.11.4
3
3
  Root-Is-Purelib: false
4
4
  Tag: py3-none-manylinux_2_39_aarch64
5
5
 
cuequivariance_ops_cu12-0.4.0.dist-info/RECORD DELETED
@@ -1,13 +0,0 @@
1
- cuequivariance_ops/VERSION,sha256=QLjrQACpE6d5EJBTXykdPTaYdBYqie88nj1OiHobnnk,6
2
- cuequivariance_ops/__init__.py,sha256=ba7jv_WICRROtLbDU2O1u0MHxp6VkVu0-UGKuQxf9iw,1255
3
- cuequivariance_ops/_version.py,sha256=o9Flao_mTq2Y7TrrjnSCqEAgebmA0sGozsl15qVI13Y,730
4
- cuequivariance_ops/common/common.hpp,sha256=2zDyE5lGugQL43vmM4_ylmp-Tz8OBFnPRsdFra_1BdM,2787
5
- cuequivariance_ops/common/nvtx.hpp,sha256=Wi6z9b-yFUNq6ShJjjcsdxQRqCygd4xGegGJrqUI9Wk,708
6
- cuequivariance_ops/equivariance/fused_tensor_product.cuh,sha256=bOXR5UWU9gNYRfdh6k28NEkV3CUU2ijmh6y7c0ND0J4,8283
7
- cuequivariance_ops/equivariance/segmented_transpose.cuh,sha256=gfSZhRBwSqwVAgFCCiGtI-NJ8yDy9tV_iCg1G2KpctY,1766
8
- cuequivariance_ops/equivariance/tensor_product_uniform_1d_jit.hh,sha256=oWhSS0ZmMHlye8eTucweoGBtzN1H0nN1GX_Rz-MsPqI,2002
9
- cuequivariance_ops/lib/libcue_ops.so,sha256=VQP3gnNy4jpVad__bbaZiakvx5c8J63Q2knrvfIRLJc,81794536
10
- cuequivariance_ops_cu12-0.4.0.dist-info/METADATA,sha256=na1Ly8dpRX4aVspiUYshcsTnjVmh0zsW13AtoteoEJ0,20842
11
- cuequivariance_ops_cu12-0.4.0.dist-info/WHEEL,sha256=teK9zuS7Jv7dMHQejkMfDwwTIgdimcBypnObHv4zSrs,116
12
- cuequivariance_ops_cu12-0.4.0.dist-info/RECORD,,
13
- cuequivariance_ops_cu12-0.4.0.dist-info/licenses/LICENSE,sha256=rvp0QV9FuOdxz_CGWTd9DgId4xh2BByyXfBBnb0ejZM,18279