sketchssm/__init__.py,sha256=RP_wk6jzhCCQNPgnBsYhSygtgXu8-xjC1nJhL21Cin0,342
sketchssm/calibration/__init__.py,sha256=B8XzRWhqHiZcMQz5CTyi60F589vZzLgmvYJWB-hiO1A,203
sketchssm/calibration/__main__.py,sha256=f8TogqYsfXZ5ZpwzHt3770HmqDJ3D1lDrCLBGTTAaT8,7082
sketchssm/calibration/bundles.py,sha256=XT3mS0eAixKrh3-Zju1VboCBgb_0u8M1iMQLnu9hkVM,3826
sketchssm/calibration/calibration.py,sha256=4NknO81nhmw4SBtHDNoEm9EN8Tond7HFaRWYbqsk6sg,11728
sketchssm/calibration/run.py,sha256=QeSChpsuO_UvDnPjDfslV_g4Oe4jPpoQtaamkuUrJ3c,6335
sketchssm/calibration/adapters/__init__.py,sha256=wyJwSnROSZcsSbbunKrcpuQYiIhxt_Mq2ajHVxUiCkw,1967
sketchssm/calibration/adapters/base.py,sha256=_YQdm7_qvfNc4tgr9SekL3fxbpbkPK4rqu6-AOkdIY8,3392
sketchssm/calibration/adapters/gdn.py,sha256=0m-gBS-WtP5AnaIvNjHjc6Ot1B56zaazhbe12uLWDiE,608
sketchssm/calibration/adapters/kda.py,sha256=Vx8wBNu0bj2lvaQuwuTxCb5swuSSl8tQ1dEJFCewYlg,640
sketchssm/calibration/adapters/mamba2.py,sha256=n4VVSPxLgNM92uoiUfpcqEdLkAL2Al36yThdK82iXa0,643
sketchssm/calibration/collection/__init__.py,sha256=Y7dozTXL0FgAQFWQzC4FHzHe1eQ1nIgEwSoH1iLoMN8,193
sketchssm/calibration/collection/config.py,sha256=nMEgcg3UeeH_0gIWBIwvABl-rtyNwP-ay7zggV6wE7E,3953
sketchssm/calibration/collection/data.py,sha256=Q8KRPGAKZWNjVbbbV2Yo9BYzT8JYzgZDqB3GZLJLY8Y,1609
sketchssm/calibration/collection/engine.py,sha256=ardNvb3M2JvJbgkrp-kM0U_sfgWcHV7Y9bgVv7klc2I,9055
sketchssm/calibration/collection/forced_tokens.py,sha256=0RhL-fFCeAI23on8-RO9uCK9q7iJ3UfdHFOqRRzmDU4,1977
sketchssm/calibration/collection/models.py,sha256=5gOknG9HQGkdsnLUh-A_cALD-Q6H3UH36hNzYAoPC4g,3175
sketchssm/calibration/collection/paired.py,sha256=J-uyYzcLrP-oLkNMump9gntfH7wkLyvKzTqS9vxPmws,8307
sketchssm/calibration/collection/pipeline.py,sha256=0Z5kGegqbg00oXmqhAuDGXlWNF_sOYoV6ayTdhbB4Vg,5502
sketchssm/calibration/collection/storage.py,sha256=fuRbkxQzKm0MzRP-tCFggMY1pZlde-xxO9q488lzHio,1341
sketchssm/calibration/collection/worker.py,sha256=Qoi47U-tB8NZwYrHVPnJ0CtDDrK6EFnpnieBxSydmPU,5110
sketchssm/calibration/collection/hooks/__init__.py,sha256=Y7dozTXL0FgAQFWQzC4FHzHe1eQ1nIgEwSoH1iLoMN8,193
sketchssm/calibration/collection/hooks/gdn.py,sha256=EWMJrZgKHtpMsXf3WtjuR0xG6wYdkaAuWhPwQmkYF7A,18518
sketchssm/calibration/collection/hooks/kda.py,sha256=UEU7uD2dSOnkwFsQP94APqFQUEiVVFtttkzY9iYGdBE,2039
sketchssm/calibration/collection/hooks/kda_statistics.py,sha256=YO03W7Z8dPfhblE6-q8jrkim-8yAiC4jDmdmsXgY4Fg,5071
sketchssm/calibration/collection/hooks/mamba2.py,sha256=ksbEWC9ubRNLEZznFXVmxzfMb1OJYUYoL23WSHblLmE,23338
sketchssm/calibration/collection/hooks/mamba_scan.py,sha256=ntLlnh6sBSUp7xLRWqnObx7RRG2k9y3BVgJiTOZf_S8,2018
sketchssm/calibration/collection/native/__init__.py,sha256=Y7dozTXL0FgAQFWQzC4FHzHe1eQ1nIgEwSoH1iLoMN8,193
sketchssm/calibration/collection/native/common.py,sha256=2jP6CgnIrMlByC6TmiwSPEziGHftFmsTAgO5yj0tBs8,3634
sketchssm/calibration/collection/native/gdn.py,sha256=YcEFineZTUWOjGHlgFwpc_2c0VtIF3-g8j8k32oGdoY,6182
sketchssm/calibration/collection/native/kda.py,sha256=NPCiwJK-eBaM1Cf5P46aTwJL7pj3FTcxtPQ_GwUVpSM,4948
sketchssm/calibration/collection/native/mamba2.py,sha256=JdViQRP-9PUIpY23jwhrLZ4JO_zMbJLmXHSj2tBPbnk,5327
sketchssm/calibration/collection/observers/__init__.py,sha256=Y7dozTXL0FgAQFWQzC4FHzHe1eQ1nIgEwSoH1iLoMN8,193
sketchssm/calibration/collection/observers/gdn.py,sha256=8aPtryLAQhN9UDZaeEflFfOSIZI9fy1-4Fh8yV-v1G4,2965
sketchssm/calibration/collection/observers/kda.py,sha256=62v-eGgv4CVewPp3DGSLzd826O8zNllGq-SNGnMr6P0,3658
sketchssm/calibration/collection/observers/mamba2.py,sha256=-VQWmdzDT24J5lOqn_pTmwbBNWs8pqom09DawIB0edc,2518
sketchssm/calibration/collection/packed/__init__.py,sha256=Y7dozTXL0FgAQFWQzC4FHzHe1eQ1nIgEwSoH1iLoMN8,193
sketchssm/calibration/collection/packed/fp4.py,sha256=wHppJuQ_W3wIkWayzHuUl6avi9KpvzTCBrjrHVfRWc4,645
sketchssm/calibration/collection/packed/glm.py,sha256=fkEXn8kFf-7t2dYsJlTTuhipWS1zLD2pQfCFvLzt5rQ,9407
sketchssm/calibration/collection/packed/qwen.py,sha256=CAWW51Ch8gXGhCGs7LmmVFM6Yh3AL11wB1wK8HDVvXE,8850
sketchssm/calibration/collection/packed/qwen3_5.py,sha256=Hdu6zc59s5hWejjj01TTlOE0CfYmFIbEXx09RXg9Pms,5762
sketchssm/calibration/collection/packed/super.py,sha256=fO9qsN7wdFUstz0bOWXZdVj2Hb8qbvveKJBQ8RDZ1P8,7904
sketchssm/calibration/core/__init__.py,sha256=gSk8dube9h4iRvRVPjYHgbkDyvU_QiVuuQas6KwnQuY,162
sketchssm/calibration/core/_solver.py,sha256=9wALXnVxO_n8654TiilRj4ahxQXuHjzRWV900IU3Jtg,3969
sketchssm/calibration/core/_statistics.py,sha256=JdUy3bEDUVbEMHUM3vQgR7xHaMAaSe1gNKlQIxle33k,3270
sketchssm/calibration/core/allocation.py,sha256=PxVpgulD-dXjSkVpk445FeTQNw4rEeDXQTfV4qClNNE,4214
sketchssm/calibration/core/basis.py,sha256=QCIf1zGtP3SUqS9hKguL9R4KnbrtXmShQmMbZhDon1s,3001
sketchssm/calibration/core/export.py,sha256=T5QxZH0ElSAb2jQ1PRc-jCpy_cEBsgoF795ct6mVabw,1773
sketchssm/calibration/core/frames.py,sha256=ueu5jxRlyPw14cpyZosVfg8bLvbVAwCFwYWDmLUIiA4,546
sketchssm/calibration/core/reuse.py,sha256=WYy2qkQ6tcIoLMGNJcilpCdRBsI_SqFN7eL5vvzIZn4,2436
sketchssm/calibration/core/scoring.py,sha256=hGRnxYjO2yWOYEds7y5SUcx90RUDLCZPUETJ6tSEezc,2808
sketchssm/calibration/core/traffic.py,sha256=3VNicZ-iOMWGWY6aXHUC6VxVYJwlE1-3juIfXtWfiAs,2561
sketchssm/calibration/tools/__init__.py,sha256=8g53uifZRbgUH0P8ZzvJehRVYM7fOmJH9Z61NEdSd70,193
sketchssm/calibration/tools/check_gdn_hooks.py,sha256=GtV1yvu59lQIrFsNuNSAIVlJ5qYFqKeJqp9S3fYYXu4,1646
sketchssm/calibration/tools/check_kda_collection.py,sha256=iMgtZoeLbujYyzYZU36ZzoikqELar1LexQ6zCr1_wBQ,2443
sketchssm/calibration/tools/check_mamba2_hooks.py,sha256=BTBrWd4EAfWHUcuXt9VO8nseBK7QsTtiS6D0nhHTDiM,1768
sketchssm/calibration/tools/make_nano_layer_subset.py,sha256=AR-1SGoRD8DE_vF4Fcrf42mGgK0tTppZOjMH2d-RGy0,3051
sketchssm/calibration/tools/make_toy_data.py,sha256=B98Tu1Q2sbgezNyGXuE05znXURi2TNLv7W5_kFzjFGc,1109
sketchssm/calibration/tools/verify_bundles.py,sha256=4Y_WzYvYvFda1h8P76D_fZiB08fnEKg8jHWpKGwM8bE,2725
sketchssm/kernels/README.md,sha256=ogaZloEgdYbtAfv_bErHrZJhOpj40xuQvjTcUxNM7wc,9954
sketchssm/kernels/__init__.py,sha256=lAt7J4UHy96YuJZ1d_dq8YMRSC5Ipfxnnh59ixNjcVM,1005
sketchssm/kernels/_driver.py,sha256=yQ52r_1QFz60BU7haoqlA1ZNetPPGMpRTfgAgXUKkWs,8364
sketchssm/kernels/_kernel.py,sha256=VxWpqOem7TEh6fEjqKOGWV57zVFxAB5QkDMH5Uir6mo,15372
sketchssm/kernels/_nvrtc.py,sha256=1o7TbxmssunxtubD2uQ9DjZcrJMiv3HJ0BwQxveGxIg,6974
sketchssm/kernels/_runtime.py,sha256=GyUkBajHuhI0QBpEAI0v_tCaX4YKcExXMAqUh7MOB4Q,6844
sketchssm/kernels/build.py,sha256=AdbA4xh9fqh2yiYxJT88xS5HCqAILZzwq5ApLi6r2vY,10496
sketchssm/kernels/gdn.py,sha256=_VuytATU0AV3NMAaT7hNRYEJwTYgOqYpT_8hIRA1J8I,24105
sketchssm/kernels/kda.py,sha256=yDudXAxE-BejsDap_TkCXu4dzNp741pJfRqto-dObsI,25959
sketchssm/kernels/mamba2.py,sha256=svxpZMO31nmWncEmjlp84uyZ2qxe8gUIbQ8X3Ehtj9U,25803
sketchssm/kernels/types.py,sha256=Bvn4ljTiACikshTWqnA_ImYPVXjBmaU0ZCUFG7SjOPw,2142
"sketchssm/kernels/configs/gdn/num_k_heads=16,num_v_heads=16,device_name=NVIDIA_B300_SXM6_AC.json",sha256=wThJVTzRgtEUk4XdZAlJLCORMbsu2F5v-9sIIRrn5sU,389
"sketchssm/kernels/configs/gdn/num_k_heads=16,num_v_heads=16,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=gTQ7GGv49_HKbDKSIjzT-fW9g_s355Mlf3z6AikchMI,111
"sketchssm/kernels/configs/gdn/num_k_heads=16,num_v_heads=32,device_name=NVIDIA_B300_SXM6_AC.json",sha256=wThJVTzRgtEUk4XdZAlJLCORMbsu2F5v-9sIIRrn5sU,389
"sketchssm/kernels/configs/gdn/num_k_heads=16,num_v_heads=32,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=hiliCX0SrsOn-cZjPBT34vtEsexjtyXW4mk_nH7_aYY,36
"sketchssm/kernels/configs/gdn/num_k_heads=16,num_v_heads=48,device_name=NVIDIA_B300_SXM6_AC.json",sha256=wThJVTzRgtEUk4XdZAlJLCORMbsu2F5v-9sIIRrn5sU,389
"sketchssm/kernels/configs/gdn/num_k_heads=16,num_v_heads=48,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=qeMpS9uGkmx9c95aHRNJx1w9HuaMVVjAiH-If8XRgEM,111
"sketchssm/kernels/configs/kda/kda,num_heads=64,device_name=NVIDIA_B300_SXM6_AC.json",sha256=__ley8DsPV43oe0TY8Xyfy6ghPAuMABdSTqEN3LZ4P4,131
"sketchssm/kernels/configs/kda/kda,num_heads=64,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=RdL-w8TvUsL1O1Il97wZv5svnyOZ4d0FyQconufE4yU,86
"sketchssm/kernels/configs/mamba2/head_dim=128,heads_per_group=16,state_size=256,device_name=NVIDIA_B300_SXM6_AC.json",sha256=Gx0xJPbJEws_6Q8-sriyQGCFyd4CppOv3QEuX7vrhpg,868
"sketchssm/kernels/configs/mamba2/head_dim=128,heads_per_group=16,state_size=256,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=9kCLLOKgNmn84Z37s5SM33RK6koTm-BmJGkhWH8WN7I,138
"sketchssm/kernels/configs/mamba2/head_dim=128,heads_per_group=24,state_size=256,device_name=NVIDIA_B300_SXM6_AC.json",sha256=Gx0xJPbJEws_6Q8-sriyQGCFyd4CppOv3QEuX7vrhpg,868
"sketchssm/kernels/configs/mamba2/head_dim=128,heads_per_group=24,state_size=256,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=zXJJGYveivJTUpyXD0zdodWRtqEOg6nsE0gAFEfYNgQ,162
"sketchssm/kernels/configs/mamba2/head_dim=64,heads_per_group=80,state_size=64,device_name=NVIDIA_B300_SXM6_AC.json",sha256=Gx0xJPbJEws_6Q8-sriyQGCFyd4CppOv3QEuX7vrhpg,868
"sketchssm/kernels/configs/mamba2/head_dim=64,heads_per_group=80,state_size=64,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=YD1b9R8EI9YhklVUiRlMFTimb6ZCkK_xbPQ_cVqfqdM,86
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,device_name=NVIDIA_B300_SXM6_AC.json",sha256=Gx0xJPbJEws_6Q8-sriyQGCFyd4CppOv3QEuX7vrhpg,868
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=piElr6r28033xh_6MJyBYAMG6rPeUVt5TetoLXmzQvw,179
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,device_name=NVIDIA_RTX_PRO_6000_Blackwell_Server_Edition.json",sha256=it32kZsA1B3GMUjmhmedEiS4IaDK32FgGwRI5lfQCfs,534
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,window=32,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=mLJ547hQXiXA0-Af7Yz9Ad9pdWiDW_0AmqbSt4HFvCA,180
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,window=64,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=bqUF3H85O9rS8ebJUf724Lip32YLD8EoxQwC4OWWFl0,157
sketchssm/kernels/csrc/gdn/gdn_sketch_flush.cuh,sha256=9QQiB11RuZDvKhYTX91IAYoYENtLAWOuR7p7UMzmbpE,52333
sketchssm/kernels/csrc/gdn/gdn_sketch_step.cuh,sha256=v13VIRAnjIpI5aWoFcG9T-Z81bDJuyQI0zGinPoF1KE,31914
sketchssm/kernels/csrc/include/array,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/map,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/stdint.h,sha256=9wTFNYDZXXAwr-aZ8NO5-ORaIoeKABrBF66qFOQqIAo,476
sketchssm/kernels/csrc/include/stdlib.h,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/type_traits,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/unordered_map,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/ATen/cuda/CUDAContext.h,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/c10/cuda/CUDAException.h,sha256=lWsxcoRTBkis7b82m2T5d8dafLuQmEaOCiP5aDdNTis,201
sketchssm/kernels/csrc/include/torch/extension.h,sha256=3d3TmYcddpTuhUdBKSacdhV52b-c05udkAaDhti4Zak,688
sketchssm/kernels/csrc/kda/kda_sketch_flush.cuh,sha256=KHWHPRm8XsJQ4K9GxFjjrknUVvgb0UgjCLQlUa_eL8w,62210
sketchssm/kernels/csrc/kda/kda_sketch_step.cuh,sha256=4uZ4TUd9PtQX9F9e6nbZoR1rG-j16PXT9pAM2iUcdV0,27386
sketchssm/kernels/csrc/mamba2/mamba2_sketch_flush.cuh,sha256=OiCORmoR6acu22_P11OiedFXO-RKQzfEGoL0tf6_A20,50891
sketchssm/kernels/csrc/mamba2/mamba2_sketch_nf.cuh,sha256=l4d3B1gNfII4w7f1Yd_QszGfNFYDE_emzAhkjZY_fdo,59063
sketchssm/kernels/tools/__init__.py,sha256=JsXLpDkXr-QFhqDIRt6xnX9fhcu7J4piMrgs_DyJltU,112
sketchssm/kernels/tools/benchmark_gdn.py,sha256=CAuL3hOvxrUC61OHucxB9sJFyCKM6_hBu_eAtGscyqI,4238
sketchssm/kernels/tools/benchmark_kda.py,sha256=Zde_VMLKrTVBJUxzUoRzaeFMqJuvayTqrTn1GorbJbs,4092
sketchssm/kernels/tools/benchmark_mamba2.py,sha256=AtyDmPsKfU_yTaCqSc4H0ZbBiZg1fEXTfamy8A51-QU,6850
sketchssm/kernels/tools/benchmark_utils.py,sha256=mL3cSrcTjKntz70aY9VbkzNSUA_sx26SHNLJzSuUWJg,10825
sketchssm/kernels/tools/tune_gdn.py,sha256=vbvt42rwMzl5VwFrMCmc_uzSq6l6mVxPS_YlJ3RwPQE,2769
sketchssm/kernels/tools/tune_kda.py,sha256=9UTJR5W-XYVtREwM6fnoclewY6e0bfv1f1xD8_DX4bU,2709
sketchssm/kernels/tools/tune_mamba2.py,sha256=VOViCbZQRMtWX6F8TIRVN-XO5Yrxhg_1h9W0LH6hB5Y,3073
sketchssm-0.1.7.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
sketchssm-0.1.7.dist-info/METADATA,sha256=HZEHSi24qw_zlKdZR7b7ZEQxmqIbtpymcmn1q2_RNhg,10010
sketchssm-0.1.7.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
sketchssm-0.1.7.dist-info/entry_points.txt,sha256=vUOvMr7VkItV8oFDGl0idANi9lemDI2xe_gGIzKCzsE,76
sketchssm-0.1.7.dist-info/top_level.txt,sha256=9-2IT4yFfOrd0Jbor7UF7ycNs9hrfDVTl5sU6a_-WY8,10
sketchssm-0.1.7.dist-info/RECORD,,
