mslk-cuda-nightly 2026.1.19__cp310-cp310-manylinux_2_28_x86_64.whl

mslk_cuda_nightly-2026.1.19.dist-info/RECORD

@@ -0,0 +1,116 @@
+mslk/__init__.py,sha256=3DUbQhu4NCLwLH9QZ6NnzxxqUj559WLBetHSBbDDkkc,1863
+mslk/mslk.so,sha256=8Zu-Y1OXiiZ9AqseaCmDT12hoqDQDVj2Sw4hXjPXwY4,263518704
+mslk/version.py,sha256=RnP_oA7cCUM0Vxq6NqiLLsPJUOlkr4gkmxhU0bk6eUs,318
+mslk/attention/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/attention/cutlass_blackwell_fmha/__init__.py,sha256=wSqB-35w4-vbjFTnZGWVpsI2IetmA1Lmmx2VIc4BHsA,1101
+mslk/attention/cutlass_blackwell_fmha/cutlass_blackwell_fmha_custom_op.py,sha256=XljnytMTfoYHl7sgaBMxLntkdr1iKmxQhz7lKzJow5U,10168
+mslk/attention/cutlass_blackwell_fmha/cutlass_blackwell_fmha_interface.py,sha256=R6ETEWnX9D6ZbzkhPNMZ0nlE-sbcU3Y8UFPBpKV2ziI,16837
+mslk/attention/flash_attn/__init__.py,sha256=AZp5t35PfljwdBgLb-Y6fTWAaCB-mhMOE5LBS0Vyp9o,440
+mslk/attention/flash_attn/ampere_helpers.py,sha256=agVjWeRumr2spTdNCZmlXqlS3PXJA1nt2K4H1GfjheU,3725
+mslk/attention/flash_attn/barrier.py,sha256=rUbB9orPXhNY_IcSzCaw0KhmZMDw3Y83K9LDQBTj5lI,2073
+mslk/attention/flash_attn/benchmark.py,sha256=YRg1tI9y7FTFnr4pIg_-mhPQV8RHWDXdj3F79E6Cy_8,7386
+mslk/attention/flash_attn/blackwell_helpers.py,sha256=eHuPwU_OChuQNcE3dsuHUhy-0UpXS1mIWVq4I-WUxSI,33394
+mslk/attention/flash_attn/block_info.py,sha256=QteXMG2GjGpefJtjriBXI1HKGElQV2kzXZkMM4qJRg4,5096
+mslk/attention/flash_attn/block_sparse_utils.py,sha256=M1mBT0cFS2hzFEscLfeBos87kDq1jaHrPsJU1-IlVRA,53771
+mslk/attention/flash_attn/block_sparsity.py,sha256=_cIknigkskyeUNbeDEfoQ6hFUiXWOi8KSzXx5TGJ3Ig,7451
+mslk/attention/flash_attn/compute_block_sparsity.py,sha256=QHzU2kT3rmwxHvBEoLP6397-HbD-Xiw3FTS5K9YIyX0,14187
+mslk/attention/flash_attn/copy_utils.py,sha256=Cq282UF63QE-6Y-_qZk_iuqI46q0xiCM-7X_QwQ2FqA,11732
+mslk/attention/flash_attn/cute_dsl_utils.py,sha256=beT0ylTPqPOb-T4MRELQI-wu2xzbfc6hbS9I6C87-dk,5228
+mslk/attention/flash_attn/fast_math.py,sha256=PTe48tmoCl_X_8wIz6yIRMmsJSLU8jE_1V3P52p3VQs,511
+mslk/attention/flash_attn/flash_bwd.py,sha256=WDh7pxbFjqqLkF0vemw4AdFKDCIBLrSDtxxhJvtxSiY,64694
+mslk/attention/flash_attn/flash_bwd_postprocess.py,sha256=xufw9vefQeV8ht0PrN1m_jCH2agymiKoa_TCJw7pdto,20723
+mslk/attention/flash_attn/flash_bwd_preprocess.py,sha256=v7NahDjT6aCs24MSai5BhMwCtyCDUvYgCpzQhsaUD9o,15722
+mslk/attention/flash_attn/flash_bwd_sm100.py,sha256=DOCUqSGtIM-kghcfG4CgetlYzUBpuJiKyVgRAyJP6vo,128240
+mslk/attention/flash_attn/flash_bwd_sm90.py,sha256=D3rycG-3myGUxm8TfmKHhbSdkf7P4CHuThXoUb-tG1I,71435
+mslk/attention/flash_attn/flash_fwd.py,sha256=UAfSEfozKSS9PKAbyycA2sLt0IeQLoHeWpX3ADRhDvQ,106397
+mslk/attention/flash_attn/flash_fwd_combine.py,sha256=kn_-BMYxFiacdIN05_-NIjoedlCBXzh5XkPc93uoG6g,31837
+mslk/attention/flash_attn/flash_fwd_sm100.py,sha256=ds9t9jyBreHTuGNzLR1dAuuw5rr-IPhZuomqXCEaBMA,127185
+mslk/attention/flash_attn/hopper_helpers.py,sha256=KkQlC5BtMuTcgOXsg-s7-lkjs3ar_J4furPy7UVXzo0,3629
+mslk/attention/flash_attn/interface.py,sha256=MEWt2w83ZxRSTHo9xm9pumLk7GNQoMIWMVeMFz4nXA8,65463
+mslk/attention/flash_attn/mask.py,sha256=-9NQtc80uU70bJ6PzdN9xkmt9f4WYCEPMOKIMOlpxQg,32291
+mslk/attention/flash_attn/mma_sm100_desc.py,sha256=7-Qqrq5Du3mrUlLOG7MhViyAqBgN-RDaWBquSNDI5L8,10896
+mslk/attention/flash_attn/named_barrier.py,sha256=w2Ki11izChKNW22y5s7jT7h7eUVpvhR2GfRoUvTDTcQ,882
+mslk/attention/flash_attn/pack_gqa.py,sha256=621o9xOMensvuB57Vk-sGLzH5g4-CuWBTQGOMWgfnmI,7472
+mslk/attention/flash_attn/paged_kv.py,sha256=wiGWFmw64GNLlhHBEitEM18IoOVE2lAGz140lAC5r5U,6380
+mslk/attention/flash_attn/pipeline.py,sha256=jniVCWLeCPxb0GZKurB75LdS4XqdmF3M3L3ms9JzLxk,10055
+mslk/attention/flash_attn/seqlen_info.py,sha256=wXx23x-1TmQDMTjcR-Ps7bofc5-Gltv1YIwFrBU2amY,4856
+mslk/attention/flash_attn/softmax.py,sha256=zXcDP2GzLtYp2aj6HCletnq0taIUe1GJZnuuPcY2fwU,25010
+mslk/attention/flash_attn/testing.py,sha256=IU2A2PPFVB02p9SNZTjUQQl-nO4Fw-bRaK5Yxkk5G7A,15415
+mslk/attention/flash_attn/tile_scheduler.py,sha256=g7M8Tmr9FwJbB_Wt8-IPouT6tgnZqzJRcqtbuevxKyY,31151
+mslk/attention/flash_attn/utils.py,sha256=C5_XGlbUB85ZRIkCKVYV3EIEMVGJDbbwuPJxiob0Lrw,32592
+mslk/attention/fmha/__init__.py,sha256=aSYPfPqyIPQviC-NMT_7mm2HKUVTk_iQl3Jvmt5ezUQ,32193
+mslk/attention/fmha/attn_bias.py,sha256=1I6s_gY5s-qmPkS8vHmF9UoFWpOcSkuvZ2IWwaCWjqY,79354
+mslk/attention/fmha/attn_bias_utils.py,sha256=x90wbEx62hKpnGTwWPKB1KSBcofYiSqxzBBk0vEX9fQ,20507
+mslk/attention/fmha/ck.py,sha256=_f45C-7FsR6fMqfBRf3FrjieSflBaZr1giyxDG3nAdo,18307
+mslk/attention/fmha/ck_decoder.py,sha256=8JAQE9CcqcMhNRRJLGi2kNsd7yl-383QJV4FwHZmAVM,5138
+mslk/attention/fmha/ck_splitk.py,sha256=GN1xAvwiKOP31TzCpq-zViUDrV_c4ncSkxnRC_bPgIw,6376
+mslk/attention/fmha/common.py,sha256=LqWFu4joKN1zMHyJ5Hbw_ihGbQT825imIOZTIVpK14I,22867
+mslk/attention/fmha/cutlass.py,sha256=D7tKi86ctjmEq-KM-Q-liUHJIQV2BmDhaSQfiQc6r6I,17202
+mslk/attention/fmha/cutlass_blackwell.py,sha256=J702M0kBEeL44Tm1WHN8lshXfIFwGqOSOT6N63oR2AQ,19045
+mslk/attention/fmha/dispatch.py,sha256=a3OzzF0vOx3i_-58V9Fydi2bfVTjPgUcruc7mEeNdMI,7796
+mslk/attention/fmha/flash.py,sha256=c0yQc35gogsxo9w6YxV-k2bzKIORp3VkSfJj0-yEgoA,30269
+mslk/attention/fmha/flash3.py,sha256=p3Yn0Q6yl92jBsBt7at8am17_O1SsVlDqdZRclN8mUc,30073
+mslk/attention/fmha/flash_mtia.py,sha256=Cqu0Yzh5ALwk95kJUrXRtzY90gCjON5iMoW_21mz-iw,7687
+mslk/attention/fmha/merge_training.py,sha256=jHxvldCuFL2OrLfb7MbfaSS_jk_KJSMxClUgUx_-YTE,6447
+mslk/attention/fmha/split_blocks_fairinternal.py,sha256=jXAk7yN5Yg5wf-5dxlmXCh12Y3q11UNE6Y3zR8dLdfE,13364
+mslk/attention/fmha/torch_attention_compat.py,sha256=XhCoBqwboIlWQAjMTGvReYJwQLeQ3JZLNV7BBBYMNYM,6968
+mslk/attention/fmha/tree_attention.py,sha256=bkSQc-IpypcVuM2tb1souccxTU5WGCqa2rsBa4AANfk,29289
+mslk/attention/fmha/triton_splitk.py,sha256=R41DKLQpH2GRWoTCAKinAkvAaqWNyingi62F2jw3F10,50130
+mslk/attention/fmha/unbind.py,sha256=-D_Nl22hVxkCVzLyfGsv5AsQJAFPksrH3Av9VuN4xro,3957
+mslk/attention/fmha/_triton/__init__.py,sha256=70dlanOLBcEjvcvR6iYGuYQ9maFszwaHn3O-dFowfoc,222
+mslk/attention/fmha/_triton/available.py,sha256=scyA582ydP98epLCZUIDUG2i6Yj6CYYBvMKys3xX_5E,1240
+mslk/attention/fmha/_triton/splitk_kernels.py,sha256=jKseHUWHwfAufnn66hZhjNDAvW2GT0BcANqOUSaN6_g,54824
+mslk/attention/fmha/_triton/vararg_kernel.py,sha256=JwV_LoeYuYVAnTM6ApLjVFbHacS51JfKPGzaDUDjQoQ,9456
+mslk/attention/fmha/utils/__init__.py,sha256=70dlanOLBcEjvcvR6iYGuYQ9maFszwaHn3O-dFowfoc,222
+mslk/attention/fmha/utils/bench.py,sha256=hbnkN2sFcmJbP66crWMv9PiYRjfgiwshoFbytwU6UjM,2574
+mslk/attention/fmha/utils/cpp_lib.py,sha256=41WGtaX0xkXqgjYq7ct-b9W9YOaPsKydDtw1yHkzUyg,4594
+mslk/attention/fmha/utils/op_common.py,sha256=_dDSWiMRyKdhQV3cX1QMsKzBRlDcFNv1EL9C_V5ZcHM,1935
+mslk/attention/gqa_attn_splitk/__init__.py,sha256=xqeqVVMAsBq6TAI18Q0DB9ekwUV7YrNBJf-dzrkCfFs,368
+mslk/bench/comm/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/bench/comm/comm_bench.py,sha256=qMUkh6M7sZ4fciui3-9NyY_ifpnpnw1D0u-VnssFltc,8464
+mslk/bench/common/__init__.py,sha256=Md3cCHD7Ano9kV15PqGbicgUO-RMdh4aVy1yKiDt_xE,208
+mslk/bench/common/utils.py,sha256=Yu8XbK0apnZ_ya3BYJ8HxPtUj5Z61i1stNBm8WWej7Y,4751
+mslk/bench/conv/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/bench/conv/conv_bench.py,sha256=tsE3Y3tR73CVC7HnKwbaO56nZzW_8ucw1Yu0sqfyu38,16579
+mslk/bench/conv/conv_ops.py,sha256=J-Md9NIjU6JrGcsrAoMcRG-ALXdQxUmcx10Z_vJ2XhE,6072
+mslk/bench/gemm/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/bench/gemm/gemm_bench.py,sha256=ea8wKTwO85LJAourQnPCCPR8U2qARPu0tCdcC-ooEnY,28418
+mslk/bench/gemm/gemm_ops.py,sha256=-ygFm5VpJMqsWAYZnKCfEDVaWZz2ada31a0adRLEgk8,103202
+mslk/bench/gemm/grouped_gemm_bias_scale_benchmark.py,sha256=mVXJRburF_WmQSDT9JqC5PB3jGGADVOlwoweCoVBoZU,5593
+mslk/bench/moe/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/bench/moe/gather_scatter_bench.py,sha256=1-jouGOietsabqyyCv_ignUupqYPeNAJRYhgNg8uRKw,12327
+mslk/bench/quantize/quantize_bench.py,sha256=unt39hk7n4HhfVIPbLu_WumCR5tLI9Hf9DH9J0fNpBw,9538
+mslk/bench/quantize/quantize_ops.py,sha256=30jTvIl618ncY9XkkXik6vjgwAxoXDKS_-ECUi8mehc,7219
+mslk/comm/__init__.py,sha256=BlZqrlrNUqwBTgA5Fmq4cewmYrvjmoK9vJBbFwKyD0o,326
+mslk/conv/__init__.py,sha256=Bn5ee_qwWTuJ4xezydXee3S2PTlQ4rthaa-AsviMEDE,327
+mslk/gemm/__init__.py,sha256=1QDl-QdQLEhi8UJtP4RPSnKrzrzhQpy9IEwCKvsc0y0,522
+mslk/gemm/triton/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/gemm/triton/fp8_gemm.py,sha256=SAfOer-fQHH_J_fq0dhtEb0vV6o5pNPGwEJCOdjS3RY,93529
+mslk/gemm/triton/grouped_gemm.py,sha256=3rsnyiVs3T7c34yPK5qIdfv97fQTWarvCeel-lQsRpo,41292
+mslk/gemm/triton/matmul_perf_model.py,sha256=c3AqDj1qFmMtWhJp1KMuKWkvEektFwAFDwC1GC71cio,7781
+mslk/gemm/triton/utils.py,sha256=UG1M-81xHTYOf8awfBsgBimZs11DYlIN8EpTxdL-mNA,4300
+mslk/kv_cache/__init__.py,sha256=IaBdKEN8laI4INPExkYyBvka5UPA9x0Dhpdl9t7O97A,335
+mslk/moe/__init__.py,sha256=OMDB0dfRWaQINgd22pRnRXEOiSrfAgoinvMSUiEZVOo,791
+mslk/moe/activation.py,sha256=ahB90VV6yVXlDNiczUMfmPxWotbY8fh3-hHM5ocGDu8,7853
+mslk/moe/gather_scatter.py,sha256=sdfOwFgcgBiDb_GrHdMgPfB0ddwYF-lKv22E7-CaWT0,20993
+mslk/moe/layers.py,sha256=X7Ivx0yS6J37CrXxElmENgmGpC1zziKS6x3j3TRE4y4,41889
+mslk/moe/shuffling.py,sha256=OT0YPYmX_RPqg8m5jyA06IBrLzMmlaUcsVnxmvP7S30,11067
+mslk/quantize/__init__.py,sha256=a4MRGasPC3oIRiMj40aNHWqDcgxNMGUlOclqVtNGbC4,335
+mslk/quantize/shuffle.py,sha256=cOGQ0sGH9myZh7keUjouvTXdes6go5p2fvm4kK7uN58,12601
+mslk/quantize/triton/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/quantize/triton/fp4_quantize.py,sha256=QjbsVgTNfTTCCodlAU8YhBagBkj4QG0XwQdFGW0unMk,224819
+mslk/quantize/triton/fp8_quantize.py,sha256=TOGNC1Fakw_TlpQbBP7MLvS6O-535IW2p-64nLTt0LI,66551
+mslk/testing/__init__.py,sha256=NNrKh5KdiYdbxOhin8x-gw_-tvcuGex2UbS_z7MV9g0,223
+mslk/testing/attributes.py,sha256=5KWS2dY2Vir2rIjd8EtsWqygcpj5FvhwXrSNsJPw0oY,1800
+mslk/testing/rocm.py,sha256=BvmW8NGgOJIkmNNsBNxJAJyNPZxZ6UFZRKVtWpQBTIg,2760
+mslk/utils/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/utils/torch/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/utils/torch/library.py,sha256=lvWQIBOsXEy6c83c0I6qiKaJFJee-95qTjYlAUDnOxU,4653
+mslk/utils/triton/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
+mslk/utils/triton/fp8_utils.py,sha256=9Rk_34SgkgWQ3jLFg4AMilnk-be9D9amF2CsIDXLAYk,2235
+mslk/utils/triton/utils.py,sha256=UG1M-81xHTYOf8awfBsgBimZs11DYlIN8EpTxdL-mNA,4300
+mslk_cuda_nightly-2026.1.19.dist-info/licenses/LICENSE,sha256=RFGcvnIte_CSzX4PomVYueOQW4j8pMDCeOJzOp9rTPk,1533
+mslk_cuda_nightly-2026.1.19.dist-info/METADATA,sha256=fkxfWejnDAE83GnWvB0vBZ5_FBka_mddS8WhY_XX2G4,3583
+mslk_cuda_nightly-2026.1.19.dist-info/WHEEL,sha256=k9CVMKlTmOLLXq_OyiiJFbPd6UKfogV4yIUezgPmplE,108
+mslk_cuda_nightly-2026.1.19.dist-info/top_level.txt,sha256=XpTLXGElEwAsc_jUVWnoCU5N4_64prVBKBkoYUbwnus,5
+mslk_cuda_nightly-2026.1.19.dist-info/RECORD,,

mslk_cuda_nightly-2026.1.19.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: skbuild 0.18.1
+Root-Is-Purelib: false
+Tag: cp310-cp310-manylinux_2_28_x86_64
+

mslk_cuda_nightly-2026.1.19.dist-info/licenses/LICENSE

@@ -0,0 +1,30 @@
+BSD License
+
+For MSLK software
+
+Copyright (c) Meta Platforms, Inc. and affiliates. All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification,
+are permitted provided that the following conditions are met:
+
+ * Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+ * Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+ * Neither the name Facebook nor the names of its contributors may be used to
+   endorse or promote products derived from this software without specific
+   prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

mslk_cuda_nightly-2026.1.19.dist-info/top_level.txt

@@ -0,0 +1 @@
+mslk