flywheel_tpu/__init__.py,sha256=vri4savMi--2ivCxFHdGIir1MFJBXojwAimGYkfStso,309
flywheel_tpu/flash_attn_interface.py,sha256=0pHz-ct9PKJXvHgd4fKCwDj17FIw_bzMgVPLU_Ro19s,48999
flywheel_tpu/tuned_block_sizes.py,sha256=-YCPe-jCmvJWrLkDxKYk2pkFf78wDzaXBIWEgqTsrmE,6728
flywheel_tpu/tuned_configs.json,sha256=9WLckhxZeQLPrEoLtnz9lHkgwP05YbESAGpvH8yYHEI,32206
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=XjXDLqLQBLbIF99adsHnKIL3EJcrknyOurX1yRlBsuc,7618
flywheel_tpu/linear_attention/gdn/memory_ref.py,sha256=OKsyjfWC262adfSic8Or0Z4eLJ6EKxZQnWIiaP69aw0,15601
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=eYEhtQ8ncdMEOkorWHpK_ws6gZ-gI5h_W86Y6PYtsWc,39857
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=uCf4T9tEEJLAVDhWYRmN5po-OML_uKi15ikNqpSXv2g,8012
flywheel_tpu/linear_attention/kda/memory_ref.py,sha256=7emv_57E0z6WnckaQvYgvXf4d62SzlnyhKoGF1_D0hI,15595
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=1Pi3lrIB5a_uDsz4iWCoKEyBdqsWF9qErHRIIzQrZHg,41466
flywheel_tpu/pallas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
flywheel_tpu/pallas/block_sizes.py,sha256=_bgT3CAn6bKdOyHi_z_srVVUCui76XTouxO2X-6xV3U,4744
flywheel_tpu/pallas/bwd_pipeline.py,sha256=YKR9cSAj2RXyyCa2eysBidyMdvnGbLt_n2DSQDBm6wQ,21473
flywheel_tpu/pallas/copy_utils.py,sha256=4I6K3NSqDCrRmD0BmNEgy1TA3nSZ8VEjf3IQtIiMbPM,6341
flywheel_tpu/pallas/flash_bwd.py,sha256=8AjJRq1pPwxTg5tM0P1T2apenzQd4WXK_FvcEmy-Xvs,33630
flywheel_tpu/pallas/flash_fwd.py,sha256=EvgTPeSuB8DKott2r5jOmJ0S1-EgOW78S26eifAGYzI,13719
flywheel_tpu/pallas/flash_fwd_kvcache.py,sha256=Gqvz7WkStAsK4mD1g85zXYiSUg4p3JKymErAvItVC64,42395
flywheel_tpu/pallas/flash_fwd_kvcache_extend.py,sha256=6Z69DmoXrRLxI6eSCgOeOlaHZwxuiI9PIRiizg9y0Sk,43518
flywheel_tpu/pallas/flash_fwd_kvcache_varlen.py,sha256=D7qMtw6TOmL4elw4U6hdrXsWtOfi_umKkyVB52PIpZ0,9547
flywheel_tpu/pallas/flash_fwd_varlen.py,sha256=dFBO7GlXm4Xu-1zKa5hZ50Wl0Acm508B1tNsxhK4xLI,3321
flywheel_tpu/pallas/fwd_pipeline.py,sha256=jr0S4PTbURBgt0okxK4pkh1tHoJwOXLVCsxF9Nwwu9M,48695
flywheel_tpu/pallas/loop_schedule.py,sha256=HVtrUUMaZhJkoCgDnRCTu_pNGIXQiPoYytw4WXS3bSs,19661
flywheel_tpu/pallas/mask.py,sha256=5CllUnUK9RBpaYZu7hAc1srJr346MC3Z36KZvKX2ns4,1237
flywheel_tpu/pallas/rotary.py,sha256=EVdG9coh5lepFX3D4VF-fo-QWz11JmJq3m-1KRWjzqs,6138
flywheel_tpu/pallas/seqlen_info.py,sha256=-S5pLo7VvbM_jfHryp3yhKEtvkncCTdaudITG14HpoE,2336
flywheel_tpu-0.1.0.dist-info/METADATA,sha256=yssHOTvLlY5h2u_ue0QYNlVDs33evWxgbHsEfsDH6Vg,30065
flywheel_tpu-0.1.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
flywheel_tpu-0.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
flywheel_tpu-0.1.0.dist-info/RECORD,,
