flywheel_tpu/__init__.py,sha256=b2N5DjChZ9o1Az945rovWH_SaYjILQkd51dzxdYM_p8,309
flywheel_tpu/flash_attn_interface.py,sha256=O7UVZrlVdU5Xpv62CliZc0s7i82hB-YXYKEnxlEAxyg,61059
flywheel_tpu/tuned_block_sizes.py,sha256=WH988oYDZyr6A7hUOfGYzzy6aGUaqiYdxQ3cbQQrSyA,7763
flywheel_tpu/tuned_configs.json,sha256=AvtFRWdEF-dOtEIit_0DphWamCbRTm3KTYF1o0VQc8E,47800
flywheel_tpu/linear_attention/__init__.py,sha256=npgA-WtO__xwwPjHfqTP6_nbIxDPNpIIJZfC9kBkaGc,325
flywheel_tpu/linear_attention/gdn/__init__.py,sha256=XlYKxA419QVEGf7AfWqsVd0MTLMBuLqa_jJtXQhYQdg,1028
flywheel_tpu/linear_attention/gdn/compute_conv1d.py,sha256=LST_lUqr0-l27d6wzF_YAZtU5Ix_xYkhey5TD22OEE8,11501
flywheel_tpu/linear_attention/gdn/compute_gdn.py,sha256=na8wNQAdJxKt-OL8LHI2GpLAlDpf9rpJ7fenq8-8Rus,50907
flywheel_tpu/linear_attention/gdn/config.py,sha256=M_Uak3U8kwYYbQR0G1gXmR_hlquymFceA-ngp6Hn2wE,11057
flywheel_tpu/linear_attention/gdn/memory_ref.py,sha256=PHwR8i34GszwRezn8ZpvA-qjK-_XySoecJyI39q-Ms8,26989
flywheel_tpu/linear_attention/gdn/metadata.py,sha256=2GC0uDgJ2ZAjgE66hjHARRjxsUnMVHHAfgPafCFa2ZY,7300
flywheel_tpu/linear_attention/gdn/vmem_ldst.py,sha256=Fn_NpbMbCKF1LqjeARsxSm-BFhjxJupDq0DXlt1El9s,8460
flywheel_tpu/linear_attention/gdn/wrapper.py,sha256=hKLQaJ787ro57zSjUxzUa5TKUTA5Zn8AFLYyph5gstA,62668
flywheel_tpu/linear_attention/kda/__init__.py,sha256=kcl6haOtmGg5gs4_-_xqRYrgz1Eof7KkhDzBkt6wmF0,1173
flywheel_tpu/linear_attention/kda/compute_conv1d.py,sha256=V1xqxfMsAgXDJTihfhvVcLyDBXAj2eJS-Qb7g2a61mQ,11501
flywheel_tpu/linear_attention/kda/compute_kda.py,sha256=CuWZmNGsQ66Y4NJILaN-eBSS7xv3wj2f55fcaa6tSSA,55458
flywheel_tpu/linear_attention/kda/config.py,sha256=um9zB3VEZcBeey-iakm9-t-oazL2zrZP-CRla81VuUQ,11450
flywheel_tpu/linear_attention/kda/memory_ref.py,sha256=4OsN7QkoOudIbQRPW7gzSUNCLukTrK8EG69QXVtLV50,26983
flywheel_tpu/linear_attention/kda/metadata.py,sha256=7yHa3JDsh9bcCukhvjDosi1xqR9QFVzYqcikRII4vwU,7300
flywheel_tpu/linear_attention/kda/vmem_ldst.py,sha256=-0aUpcx23aTEHqC8gDk3c9I5vynHFnj-5oNgQGmvwFc,8859
flywheel_tpu/linear_attention/kda/wrapper.py,sha256=nDqIasWH5Khs65lAfM9rwSF1fNTcp7eVrDxSlL9npI4,63962
flywheel_tpu/pallas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
flywheel_tpu/pallas/block_sizes.py,sha256=inp6mMuJqPbMeMUhKPKsyscdcwUhXRs00fxrH1i4tNs,7208
flywheel_tpu/pallas/bwd_pipeline.py,sha256=VOCXEGay7vU_AY_AaXjjzuNUYsKSb8QSoXlTO8Haypw,21888
flywheel_tpu/pallas/copy_utils.py,sha256=Rd_ARyX1GH5_g0153IizqWO06X8Nh9-QlmqtPy7IbeU,10338
flywheel_tpu/pallas/flash_bwd.py,sha256=IP0AadiLHttfYfU7bYdAAmktBD5UuM5GynYVPNNVP_Y,34051
flywheel_tpu/pallas/flash_fwd.py,sha256=EvgTPeSuB8DKott2r5jOmJ0S1-EgOW78S26eifAGYzI,13719
flywheel_tpu/pallas/flash_fwd_kvcache.py,sha256=6zcEM9BxgFAS-S6z-YHAOcmBGI3kQ0qwSUHXOLtdi90,51214
flywheel_tpu/pallas/flash_fwd_kvcache_varlen.py,sha256=tDQpOE4hi5Das6wqzD9AZ4kWT8e2Lre_5sQQfJMGNGk,8621
flywheel_tpu/pallas/flash_fwd_varlen.py,sha256=2i9NTSoFpfBMFijT6pi67Rcn5p0w6Yi6XPPzNbPkTzU,3138
flywheel_tpu/pallas/flash_fwd_varlen_paged.py,sha256=dlGyMX_PKkbIunZr8Gl2ycse4-2zKmjQysgkP3Jca0M,17462
flywheel_tpu/pallas/fwd_pipeline.py,sha256=iiwsrcp9vywcdjxI68o6Kvk-dNrSJMRhx2Rskm31xso,73841
flywheel_tpu/pallas/kv_cache_write.py,sha256=3wSHnLKUKrizVyE7ryHbq_69yDMtp1FNNgoDFv7NiMY,8418
flywheel_tpu/pallas/loop_schedule.py,sha256=SAKdUbKQkRY3-9mQK0mED5LtKrMGwgHNGRDUV01663U,27707
flywheel_tpu/pallas/mask.py,sha256=5CllUnUK9RBpaYZu7hAc1srJr346MC3Z36KZvKX2ns4,1237
flywheel_tpu/pallas/rotary.py,sha256=9KLf-OfssVEu_QNdweINCGrLuigdOKJjz8V0jRyp2wY,7054
flywheel_tpu/pallas/seqlen_info.py,sha256=0Hi7dtUlib0ell45Wy_L1urO7ACwDjYrPGDyO68lzdM,819
flywheel_tpu-0.3.0.dist-info/METADATA,sha256=GVDhtstOaMA9By86a3TgfqmVMoalqFQyaY-lC5XXDR0,33743
flywheel_tpu-0.3.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
flywheel_tpu-0.3.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
flywheel_tpu-0.3.0.dist-info/RECORD,,
