flywheel_tpu/__init__.py,sha256=1egECmSkfr3Nn-PeR_R8gGedQNwsxy2oJilZhC-fFcM,309
flywheel_tpu/flash_attn_interface.py,sha256=D41fOVXOHnEQE2be048-NKPUoj7WvU17upz0qxJMZi0,58096
flywheel_tpu/tuned_block_sizes.py,sha256=3ZdbREYdwYSNfwOTB_pkDvN83xHpYzlt7oiiVVR5AyM,6886
flywheel_tpu/tuned_configs.json,sha256=tpIDTd7N4dXP9DmGipBdiTVeHiX8ILDeKldftksDOyE,35288
flywheel_tpu/linear_attention/__init__.py,sha256=brW-TO_JOPmrcnAxbJ5WGrOOdfYstTL1ygZPB-MeFqU,202
flywheel_tpu/linear_attention/gdn/__init__.py,sha256=XlYKxA419QVEGf7AfWqsVd0MTLMBuLqa_jJtXQhYQdg,1028
flywheel_tpu/linear_attention/gdn/compute_conv1d.py,sha256=enudElAcqAx82I83q_gONdSGGK_-gtPv3ynm6nsddHQ,11956
flywheel_tpu/linear_attention/gdn/compute_gdn.py,sha256=DsvhQYIVWn696EFwnX5jdI5cZE4YVe8svSVDuU_S7ZQ,42075
flywheel_tpu/linear_attention/gdn/config.py,sha256=3bfR5THJVR7IITRGka2OC9PNnJgAdN2kNGKoenRLZ7c,8181
flywheel_tpu/linear_attention/gdn/memory_ref.py,sha256=CVxpjhrrfH5VlERp_7sIr-gtlamPG-gn83ODNxF5tpw,16518
flywheel_tpu/linear_attention/gdn/metadata.py,sha256=x0WNs95p5YVfAoab-qDJFNJNo3Gvs7PlZsGAWBAbnX8,5777
flywheel_tpu/linear_attention/gdn/vmem_ldst.py,sha256=i6OWWFAhem80-hT_6Y3y727oGbI5SBrWSP3Dl8_R8mc,7548
flywheel_tpu/linear_attention/gdn/wrapper.py,sha256=Vu7sxgOA29yJSRv715QXWnIBACzY4sA_KR0whcmHdNs,41898
flywheel_tpu/linear_attention/kda/__init__.py,sha256=kcl6haOtmGg5gs4_-_xqRYrgz1Eof7KkhDzBkt6wmF0,1173
flywheel_tpu/linear_attention/kda/compute_conv1d.py,sha256=1Kb69HglIJaZwrdktSmrFHBPJVmf6U4-NfMMGRjoJpI,11956
flywheel_tpu/linear_attention/kda/compute_kda.py,sha256=YGcFTipWeCYMYW1styVvYBrR873_Ai0opeQfT-hCK7I,47214
flywheel_tpu/linear_attention/kda/config.py,sha256=7zZmzUse8v8egVm19KnkIu0y_OCBx8GL55lqv6ekRBI,8575
flywheel_tpu/linear_attention/kda/memory_ref.py,sha256=EIY6xHZwrVJpjwdcbcMVaanR-3D2KdRd1ojxx50-S2w,16512
flywheel_tpu/linear_attention/kda/metadata.py,sha256=YAENDdC6Rlo4XtmTnoZldArJXcla4DBAcGv0XkGkTFM,5777
flywheel_tpu/linear_attention/kda/vmem_ldst.py,sha256=fYG_i3C-hHTO5Zq12shZN2YaPlrl1RHPRsZai4h7rWA,7947
flywheel_tpu/linear_attention/kda/wrapper.py,sha256=jjzBt6ONLdWPfP4yqQh6c3hp1GCXMFQO4mIaXs1zH6M,43514
flywheel_tpu/pallas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
flywheel_tpu/pallas/block_sizes.py,sha256=LijgqDZ0OIFBimOLgPXWWIZCXBi_5Bf7A28rYCHSK4s,7186
flywheel_tpu/pallas/bwd_pipeline.py,sha256=ZkhcCXdZmj7iUxZSLZIuJdezu9P9qPCOD1q0WKcaS8w,21488
flywheel_tpu/pallas/copy_utils.py,sha256=Rd_ARyX1GH5_g0153IizqWO06X8Nh9-QlmqtPy7IbeU,10338
flywheel_tpu/pallas/flash_bwd.py,sha256=MaOq3j-MS3_gq7mmBGFp5glaUOBIxzhQsTyNMNmHCNw,33809
flywheel_tpu/pallas/flash_fwd.py,sha256=EvgTPeSuB8DKott2r5jOmJ0S1-EgOW78S26eifAGYzI,13719
flywheel_tpu/pallas/flash_fwd_kvcache.py,sha256=m2QZ1F4zoW2DI3Z0MiTTgkcxgyC36MyjWqX0OqoLgpY,46528
flywheel_tpu/pallas/flash_fwd_kvcache_varlen.py,sha256=6xGJbi-0E0N2Kzcqhuzn7bsjkH3YjGhfMEMQqbpEjcA,8456
flywheel_tpu/pallas/flash_fwd_varlen.py,sha256=2i9NTSoFpfBMFijT6pi67Rcn5p0w6Yi6XPPzNbPkTzU,3138
flywheel_tpu/pallas/flash_fwd_varlen_paged.py,sha256=EexebYQ_jflXr9uFGTkabEYwCtxud8uCSY6YOHpnB8Q,16435
flywheel_tpu/pallas/fwd_pipeline.py,sha256=834a_ny1BroV2CBfs5_mdOG5IcThifeDfQ7sXclYkW0,70224
flywheel_tpu/pallas/kv_cache_write.py,sha256=3wSHnLKUKrizVyE7ryHbq_69yDMtp1FNNgoDFv7NiMY,8418
flywheel_tpu/pallas/loop_schedule.py,sha256=kAbh85Oasl4QMiSMaPqaJesP_04v0bq6nzXUkc2u5GI,24112
flywheel_tpu/pallas/mask.py,sha256=5CllUnUK9RBpaYZu7hAc1srJr346MC3Z36KZvKX2ns4,1237
flywheel_tpu/pallas/rotary.py,sha256=JZRZlVAF8DTSZhHmNQy7zEcIsDSNC20bM_iMJ0tdh4s,6455
flywheel_tpu/pallas/seqlen_info.py,sha256=0Hi7dtUlib0ell45Wy_L1urO7ACwDjYrPGDyO68lzdM,819
flywheel_tpu-0.2.0.dist-info/METADATA,sha256=d8sD48IjMHj5nmQt7Vd5rD_esDBnPfOUe5x5dEwfI9s,32576
flywheel_tpu-0.2.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
flywheel_tpu-0.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
flywheel_tpu-0.2.0.dist-info/RECORD,,
