sketchssm/__init__.py,sha256=TwX539u_3afoVbBkyqszc6XB5e55B7zsAgRSOSIA99Y,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=lopsl8-XZ4w88fsE8INyPRHW2O5l0xRZ6oz_H3myFY0,2895
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=wShIvRStNaeJ8NJy7FCYIhA03Z3JP7g6Iq7bvyDmES4,10426
sketchssm/kernels/__init__.py,sha256=JXOZCgUBl9hWU-YQW4bVOET-n48J1xjnDHFuifdipIg,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=rQaWKLFiGvast5JKZem9kY5PL2105J_y7fCvpRMPvIM,29578
sketchssm/kernels/mamba2.py,sha256=93IGMIm1SPMDTao-pSQrR1rWV8hn90ydgbRrt3BJHBA,30442
sketchssm/kernels/types.py,sha256=SMcBK6cmzQOjG0ezsHXZxSnD8IviIkVIgfu_oUnGCsg,2655
"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=G3WyvhbT-zuNjRAXZc-y5-z9DzqiUJI2WSt9GFC3Brk,829
"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=G3WyvhbT-zuNjRAXZc-y5-z9DzqiUJI2WSt9GFC3Brk,829
"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=16,device_name=NVIDIA_B300_SXM6_AC.json",sha256=ciskX329mj7O4k2wYgkfTpkb-vIPL5DneisU4SqTEjM,150
"sketchssm/kernels/configs/mamba2/head_dim=64,heads_per_group=80,state_size=64,device_name=NVIDIA_B300_SXM6_AC.json",sha256=G3WyvhbT-zuNjRAXZc-y5-z9DzqiUJI2WSt9GFC3Brk,829
"sketchssm/kernels/configs/mamba2/head_dim=64,heads_per_group=80,state_size=64,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=RmqVSgoYu0_ymt8AL5_cQqmal4k7BDlBMphWO0Opc_c,63
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,device_name=NVIDIA_B300_SXM6_AC.json",sha256=G3WyvhbT-zuNjRAXZc-y5-z9DzqiUJI2WSt9GFC3Brk,829
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=cLU7Z_edPAuY_M5HnwUArXOsAfwZYKwaNTAsQnAiu4w,155
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,device_name=NVIDIA_RTX_PRO_6000_Blackwell_Server_Edition.json",sha256=aKfn3jNCPjz_s0F620WZ_94lXPuHbaRXPzdoFidFWAI,495
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,window=32,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=GK0by9vCZEO8v7wMFyR7ZuZ_-P_WcfOXcje1qvIWnZM,156
"sketchssm/kernels/configs/mamba2/head_dim=80,heads_per_group=16,window=64,device_name=NVIDIA_H100_80GB_HBM3.json",sha256=lLQTwfPWEIb6BOzDKXBc4ut36ZKILHIl5tFM576JBXY,133
sketchssm/kernels/csrc/gdn/gdn_sketch_flush.cuh,sha256=SdBB07bP4Wj07fHmwZ3RRo5Mz5nPzakL6p-o68pzZsc,53301
sketchssm/kernels/csrc/gdn/gdn_sketch_step.cuh,sha256=MylvfNGuakc76H4Hkm-RxGvLtrU3lg0URGIvvfbCoxo,33042
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=JZZDaCya2PTDAeOPdciiEjLIMdxjnzmE7J-pq3PmfXE,71919
sketchssm/kernels/csrc/kda/kda_sketch_step.cuh,sha256=taR1avk0412CSauzR118WCaTb3ng4DqaYEFl7chLf_g,30564
sketchssm/kernels/csrc/mamba2/mamba2_sketch_flush.cuh,sha256=F4vswvKl0YEM0jSvQ6KCbDrnAoqIwauo-AqHd19R7S8,60593
sketchssm/kernels/csrc/mamba2/mamba2_sketch_nf.cuh,sha256=_AmXt7ATEy3JHG2UASFF7Vj_EjLlctDvcoIIYkG7Xik,58719
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=I8LEGxzhFVlao8b9-ZgHQ-YqhKM0bFRBn7C7df-jyeo,6828
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=if8EEXW_nPBOZGN3swtPsGFIvSyT30F0Y0gazZV0M24,3177
sketchssm-0.1.9.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
sketchssm-0.1.9.dist-info/METADATA,sha256=Z8BhwqWJK6F_3X83m5aFHLMYPGMsZityl-NnnEs_W2g,9668
sketchssm-0.1.9.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
sketchssm-0.1.9.dist-info/entry_points.txt,sha256=vUOvMr7VkItV8oFDGl0idANi9lemDI2xe_gGIzKCzsE,76
sketchssm-0.1.9.dist-info/top_level.txt,sha256=9-2IT4yFfOrd0Jbor7UF7ycNs9hrfDVTl5sU6a_-WY8,10
sketchssm-0.1.9.dist-info/RECORD,,
