Skip to content
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
90 changes: 90 additions & 0 deletions aiter/configs/a8w8_bpreshuffle_tuned_gemm.csv
Original file line number Diff line number Diff line change
Expand Up @@ -597,3 +597,93 @@ gfx1250,256,4096,32768,512,torch.float8_e4m3fn,flydsl,584,0,74.9556,flydsl_bpres
gfx1250,256,8192,32768,512,torch.float8_e4m3fn,flydsl,584,0,137.2238,flydsl_bpreshuffle_wmma_t256x128x128_mw2_nw1_nb3_sk1_cm1_cn1,2003.14,4065.2,0.0
gfx1250,256,16384,32768,512,torch.float8_e4m3fn,flydsl,584,0,260.1697,flydsl_bpreshuffle_wmma_t256x128x128_mw2_nw1_nb3_sk1_cm1_cn1,2113.07,4223.81,0.0
gfx1250,256,32768,32768,512,torch.float8_e4m3fn,flydsl,584,0,505.4037,flydsl_bpreshuffle_wmma_t256x128x128_mw2_nw1_nb3_sk1_cm1_cn1,2175.51,4315.44,0.0
gfx1250,256,1,2112,7168,torch.float8_e4m3fn,flydsl,40,0,6.0724,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,4.99,2494.93,0.0066
gfx1250,256,1,3584,7168,torch.float8_e4m3fn,flydsl,40,0,6.2224,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,8.26,4130.95,0.0075
gfx1250,256,1,7168,1024,torch.float8_e4m3fn,flydsl,20,0,3.6692,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,4.0,2004.63,0.0
gfx1250,256,1,7168,1792,torch.float8_e4m3fn,flydsl,20,0,4.4383,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,5.79,2897.77,0.0
gfx1250,256,1,12288,1536,torch.float8_e4m3fn,flydsl,28,0,4.1035,flydsl_bpreshuffle_wmma_t16x64x512_mw1_nw4_nb3_sk1_cm1_cn1,9.2,4605.94,0.0
gfx1250,256,2,2112,7168,torch.float8_e4m3fn,flydsl,40,0,6.0565,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,10.0,2503.36,0.0083
gfx1250,256,2,3584,7168,torch.float8_e4m3fn,flydsl,40,0,6.2271,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,16.5,4130.14,0.0098
gfx1250,256,2,7168,1024,torch.float8_e4m3fn,flydsl,20,0,3.7166,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,7.9,1983.2,0.0
gfx1250,256,2,7168,1792,torch.float8_e4m3fn,flydsl,20,0,4.513,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,11.38,2853.38,0.0
gfx1250,256,2,12288,1536,torch.float8_e4m3fn,flydsl,28,0,4.1548,flydsl_bpreshuffle_wmma_t16x64x512_mw1_nw4_nb3_sk1_cm1_cn1,18.17,4555.36,0.0
gfx1250,256,4,2112,7168,torch.float8_e4m3fn,flydsl,40,0,6.0783,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,19.93,2498.13,0.0086
gfx1250,256,4,3584,7168,torch.float8_e4m3fn,flydsl,40,0,6.306,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,32.59,4083.01,0.0084
gfx1250,256,4,7168,1024,torch.float8_e4m3fn,flydsl,20,0,3.7272,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,15.75,1985.8,0.0
gfx1250,256,4,7168,1792,torch.float8_e4m3fn,flydsl,20,0,4.4972,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,22.85,2870.58,0.0
gfx1250,256,4,12288,1536,torch.float8_e4m3fn,flydsl,28,0,4.1863,flydsl_bpreshuffle_wmma_t16x64x512_mw1_nw4_nb3_sk1_cm1_cn1,36.07,4533.55,0.0
gfx1250,256,8,2112,7168,torch.float8_e4m3fn,flydsl,40,0,6.1235,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,39.56,2487.13,0.0098
gfx1250,256,8,3584,7168,torch.float8_e4m3fn,flydsl,40,0,6.328,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,64.96,4077.88,0.0091
gfx1250,256,8,7168,1024,torch.float8_e4m3fn,flydsl,20,0,3.7676,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,31.17,1980.81,0.0
gfx1250,256,8,7168,1792,torch.float8_e4m3fn,flydsl,20,0,4.5666,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,45.01,2841.08,0.0
gfx1250,256,8,12288,1536,torch.float8_e4m3fn,flydsl,28,0,4.2101,flydsl_bpreshuffle_wmma_t16x64x512_mw1_nw4_nb3_sk1_cm1_cn1,71.73,4532.73,0.0
gfx1250,256,16,3584,7168,torch.float8_e4m3fn,flydsl,40,0,6.409,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,128.27,4044.23,0.0096
gfx1250,256,16,7168,1024,torch.float8_e4m3fn,flydsl,64,0,3.8085,flydsl_bpreshuffle_wmma_t16x128x256_mw1_nw4_nb4_sk1_cm1_cn1,61.67,1991.81,0.0
gfx1250,256,16,7168,1792,torch.float8_e4m3fn,flydsl,20,0,4.6238,flydsl_bpreshuffle_wmma_t16x64x256_mw1_nw4_nb4_sk1_cm1_cn1,88.9,2833.84,0.0
gfx1250,256,16,12288,1536,torch.float8_e4m3fn,flydsl,72,0,4.3024,flydsl_bpreshuffle_wmma_t16x128x512_mw1_nw4_nb3_sk1_cm1_cn1,140.38,4484.05,0.0
gfx1250,256,32,3584,7168,torch.float8_e4m3fn,flydsl,40,0,6.8857,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,238.78,3797.56,0.0096
gfx1250,256,32,7168,1024,torch.float8_e4m3fn,flydsl,340,0,3.8819,flydsl_bpreshuffle_wmma_t32x128x256_mw1_nw4_nb4_sk1_cm1_cn1,121.01,2017.45,0.0
gfx1250,256,32,7168,1792,torch.float8_e4m3fn,flydsl,224,0,4.8013,flydsl_bpreshuffle_wmma_t32x64x256_mw2_nw2_nb4_sk1_cm1_cn1,171.22,2782.82,0.0
gfx1250,256,32,12288,1536,torch.float8_e4m3fn,flydsl,360,0,4.5135,flydsl_bpreshuffle_wmma_t32x128x512_mw1_nw4_nb3_sk1_cm1_cn1,267.63,4366.89,0.0
gfx1250,256,48,2112,7168,torch.float8_e4m3fn,flydsl,40,0,6.6046,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,220.05,2374.96,0.0099
gfx1250,256,48,3584,7168,torch.float8_e4m3fn,flydsl,40,0,7.301,flydsl_bpreshuffle_wmma_t16x64x1024_mw1_nw4_nb3_sk1_cm1_cn1,337.8,3612.96,0.0096
gfx1250,256,48,7168,1024,torch.float8_e4m3fn,flydsl,68,0,3.9784,flydsl_bpreshuffle_wmma_t16x128x512_mw1_nw4_nb2_sk1_cm1_cn1,177.12,2030.29,0.0
gfx1250,256,48,7168,1792,torch.float8_e4m3fn,flydsl,340,0,4.9676,flydsl_bpreshuffle_wmma_t32x128x256_mw1_nw4_nb4_sk1_cm1_cn1,248.23,2741.61,0.0
gfx1250,256,48,12288,1536,torch.float8_e4m3fn,flydsl,780,0,4.6805,flydsl_bpreshuffle_wmma_t64x128x512_mw1_nw4_nb3_sk1_cm1_cn1,387.13,4300.34,0.0
gfx1250,256,64,3584,7168,torch.float8_e4m3fn,flydsl,264,0,7.662,flydsl_bpreshuffle_wmma_t32x64x1024_mw2_nw2_nb3_sk1_cm1_cn1,429.17,3472.67,0.0097
gfx1250,256,64,7168,1024,torch.float8_e4m3fn,flydsl,748,0,4.0259,flydsl_bpreshuffle_wmma_t64x128x256_mw1_nw4_nb4_sk1_cm1_cn1,233.37,2067.38,0.0
gfx1250,256,64,7168,1792,torch.float8_e4m3fn,flydsl,340,0,5.0132,flydsl_bpreshuffle_wmma_t32x128x256_mw1_nw4_nb4_sk1_cm1_cn1,327.97,2768.14,0.0
gfx1250,256,64,12288,1536,torch.float8_e4m3fn,flydsl,780,0,4.7195,flydsl_bpreshuffle_wmma_t64x128x512_mw1_nw4_nb3_sk1_cm1_cn1,511.9,4353.33,0.0
gfx1250,256,128,3584,7168,torch.float8_e4m3fn,flydsl,750,0,8.9002,flydsl_bpreshuffle_wmma_t64x128x256_mw1_nw4_nb4_sk4_cm1_cn1,738.93,3092.64,0.0098
gfx1250,256,128,7168,1024,torch.float8_e4m3fn,flydsl,748,0,4.2482,flydsl_bpreshuffle_wmma_t64x128x256_mw1_nw4_nb4_sk1_cm1_cn1,442.32,2190.6,0.0
gfx1250,256,128,7168,1792,torch.float8_e4m3fn,flydsl,748,0,5.3828,flydsl_bpreshuffle_wmma_t64x128x256_mw1_nw4_nb4_sk1_cm1_cn1,610.9,2769.83,0.0
gfx1250,256,128,12288,1536,torch.float8_e4m3fn,flydsl,780,0,5.529,flydsl_bpreshuffle_wmma_t64x128x512_mw1_nw4_nb3_sk1_cm1_cn1,873.91,4018.21,0.0
gfx1250,256,256,3584,7168,torch.float8_e4m3fn,flydsl,781,0,11.4073,flydsl_bpreshuffle_wmma_t64x128x512_mw1_nw4_nb3_sk2_cm1_cn1,1153.06,2573.8,0.0095
gfx1250,256,256,7168,1024,torch.float8_e4m3fn,flydsl,1156,0,4.8612,flydsl_bpreshuffle_wmma_t128x128x512_mw2_nw2_nb2_sk1_cm1_cn1,773.08,2318.81,0.0
gfx1250,256,256,7168,1792,torch.float8_e4m3fn,flydsl,748,0,6.1838,flydsl_bpreshuffle_wmma_t64x128x256_mw1_nw4_nb4_sk1_cm1_cn1,1063.53,2744.89,0.0
gfx1250,256,256,12288,1536,torch.float8_e4m3fn,flydsl,1120,0,6.7635,flydsl_bpreshuffle_wmma_t128x128x256_mw1_nw4_nb3_sk1_cm1_cn1,1428.8,3778.97,0.0
gfx1250,256,512,3584,7168,torch.float8_e4m3fn,flydsl,780,0,14.1744,flydsl_bpreshuffle_wmma_t64x128x512_mw1_nw4_nb3_sk1_cm1_cn1,1855.93,2330.27,0.0097
gfx1250,256,512,7168,1024,torch.float8_e4m3fn,flydsl,1120,0,5.803,flydsl_bpreshuffle_wmma_t128x128x256_mw1_nw4_nb3_sk1_cm1_cn1,1295.23,2620.08,0.0
gfx1250,256,512,7168,1792,torch.float8_e4m3fn,flydsl,1132,0,7.5924,flydsl_bpreshuffle_wmma_t128x128x256_mw2_nw2_nb3_sk1_cm1_cn1,1732.43,2779.44,0.0
gfx1250,256,512,12288,1536,torch.float8_e4m3fn,flydsl,1376,0,8.7695,flydsl_bpreshuffle_wmma_t256x128x256_mw2_nw2_nb3_sk1_cm1_cn1,2203.93,3676.8,0.0
gfx1250,256,1024,3584,7168,torch.float8_e4m3fn,flydsl,1136,0,20.349,flydsl_bpreshuffle_wmma_t128x128x256_mw2_nw2_nb4_sk1_cm1_cn1,2585.55,1983.89,0.0096
gfx1250,256,1024,7168,1024,torch.float8_e4m3fn,flydsl,1376,0,7.6294,flydsl_bpreshuffle_wmma_t256x128x256_mw2_nw2_nb3_sk1_cm1_cn1,1970.32,3023.65,0.0
gfx1250,256,1024,7168,1792,torch.float8_e4m3fn,flydsl,1232,0,10.0397,flydsl_bpreshuffle_wmma_t128x256x256_mw1_nw4_nb3_sk1_cm1_cn1,2620.27,2924.4,0.0
gfx1250,256,1024,12288,1536,torch.float8_e4m3fn,flydsl,1252,0,12.7075,flydsl_bpreshuffle_wmma_t128x384x128_mw1_nw4_nb4_sk1_cm1_cn1,3041.88,3589.46,0.0
gfx1250,256,2048,3584,7168,torch.float8_e4m3fn,flydsl,1240,0,27.195,flydsl_bpreshuffle_wmma_t128x256x256_mw2_nw2_nb3_sk1_cm1_cn1,3869.34,2024.28,0.0096
gfx1250,256,2048,7168,1024,torch.float8_e4m3fn,flydsl,1412,0,11.1132,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,2705.32,3491.1,0.0
gfx1250,256,2048,7168,1792,torch.float8_e4m3fn,flydsl,1412,0,15.1148,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,3480.92,3035.12,0.0
gfx1250,256,2048,12288,1536,torch.float8_e4m3fn,flydsl,1252,0,22.9798,flydsl_bpreshuffle_wmma_t128x384x128_mw1_nw4_nb4_sk1_cm1_cn1,3364.23,3148.49,0.0
gfx1250,256,4096,3584,7168,torch.float8_e4m3fn,flydsl,1412,0,42.656,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,4933.73,1978.86,0.0097
gfx1250,256,4096,7168,1024,torch.float8_e4m3fn,flydsl,1412,0,20.3685,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,2952.08,3449.18,0.0
gfx1250,256,4096,7168,1792,torch.float8_e4m3fn,flydsl,1412,0,28.292,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,3719.31,2788.96,0.0
gfx1250,256,4096,12288,1536,torch.float8_e4m3fn,flydsl,1416,0,39.7741,flydsl_bpreshuffle_wmma_t256x256x256_mw2_nw2_nb2_sk1_cm1_cn1,3887.42,3163.59,0.0
gfx1250,256,8192,3584,7168,torch.float8_e4m3fn,flydsl,1412,0,82.2675,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,5116.32,1739.82,0.0097
gfx1250,256,8192,7168,1024,torch.float8_e4m3fn,flydsl,1412,0,38.1558,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,3151.79,3490.14,0.0
gfx1250,256,8192,7168,1792,torch.float8_e4m3fn,flydsl,1412,0,51.8032,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,4062.56,2798.39,0.0
gfx1250,256,8192,12288,1536,torch.float8_e4m3fn,flydsl,1416,0,73.3097,flydsl_bpreshuffle_wmma_t256x256x256_mw2_nw2_nb2_sk1_cm1_cn1,4218.24,3175.35,0.0
gfx1250,256,16384,3584,7168,torch.float8_e4m3fn,flydsl,1412,0,158.2093,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,5320.89,1647.0,0.0097
gfx1250,256,16384,7168,1024,torch.float8_e4m3fn,flydsl,1412,0,66.0897,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,3639.27,3918.89,0.0
gfx1250,256,16384,7168,1792,torch.float8_e4m3fn,flydsl,1412,0,90.8838,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,4631.26,3048.8,0.0
gfx1250,256,16384,12288,1536,torch.float8_e4m3fn,flydsl,1412,0,140.3126,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,4407.84,3183.56,0.0
gfx1250,256,32768,3584,7168,torch.float8_e4m3fn,flydsl,1412,0,307.5011,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,5475.19,1611.22,0.0097
gfx1250,256,32768,7168,1024,torch.float8_e4m3fn,flydsl,1412,0,126.7929,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,3793.87,4027.49,0.0
gfx1250,256,32768,7168,1792,torch.float8_e4m3fn,flydsl,1412,0,177.297,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,4748.04,3053.22,0.0
gfx1250,256,32768,12288,1536,torch.float8_e4m3fn,flydsl,1412,0,273.1677,flydsl_bpreshuffle_wmma_t256x256x128_mw2_nw2_nb4_sk1_cm1_cn1,4528.17,3201.38,0.0
gfx1250,256,1,2048,512,torch.float8_e4m3fn,flydsl,8,0,3.3917,flydsl_bpreshuffle_wmma_t16x64x128_mw1_nw4_nb4_sk1_cm1_cn1,0.62,310.52,0.0
gfx1250,256,2,2048,512,torch.float8_e4m3fn,flydsl,8,0,3.4198,flydsl_bpreshuffle_wmma_t16x64x128_mw1_nw4_nb4_sk1_cm1_cn1,1.23,309.31,0.0
gfx1250,256,4,2048,512,torch.float8_e4m3fn,flydsl,8,0,3.4313,flydsl_bpreshuffle_wmma_t16x64x128_mw1_nw4_nb4_sk1_cm1_cn1,2.44,310.96,0.0
gfx1250,256,8,2048,512,torch.float8_e4m3fn,flydsl,8,0,3.4614,flydsl_bpreshuffle_wmma_t16x64x128_mw1_nw4_nb4_sk1_cm1_cn1,4.85,313.58,0.0
gfx1250,256,16,2048,512,torch.float8_e4m3fn,flydsl,8,0,3.4945,flydsl_bpreshuffle_wmma_t16x64x128_mw1_nw4_nb4_sk1_cm1_cn1,9.6,321.16,0.0
gfx1250,256,32,2048,512,torch.float8_e4m3fn,flydsl,8,0,3.4612,flydsl_bpreshuffle_wmma_t16x64x128_mw1_nw4_nb4_sk1_cm1_cn1,19.39,345.55,0.0
gfx1250,256,48,2048,512,torch.float8_e4m3fn,flydsl,52,0,3.4526,flydsl_bpreshuffle_wmma_t16x128x128_mw1_nw4_nb4_sk1_cm1_cn1,29.16,367.77,0.0
gfx1250,256,64,2048,512,torch.float8_e4m3fn,flydsl,52,0,3.5116,flydsl_bpreshuffle_wmma_t16x128x128_mw1_nw4_nb4_sk1_cm1_cn1,38.22,382.59,0.0
gfx1250,256,128,2048,512,torch.float8_e4m3fn,flydsl,52,0,3.5095,flydsl_bpreshuffle_wmma_t16x128x128_mw1_nw4_nb4_sk1_cm1_cn1,76.49,466.85,0.0
gfx1250,256,256,2048,512,torch.float8_e4m3fn,flydsl,332,0,3.5572,flydsl_bpreshuffle_wmma_t32x128x256_mw1_nw4_nb2_sk1_cm1_cn1,150.93,626.4,0.0
gfx1250,256,512,2048,512,torch.float8_e4m3fn,flydsl,712,0,3.6951,flydsl_bpreshuffle_wmma_t64x128x128_mw1_nw4_nb4_sk1_cm1_cn1,290.59,922.27,0.0
gfx1250,256,1024,2048,512,torch.float8_e4m3fn,flydsl,712,0,3.957,flydsl_bpreshuffle_wmma_t64x128x128_mw1_nw4_nb4_sk1_cm1_cn1,542.7,1457.46,0.0
gfx1250,256,2048,2048,512,torch.float8_e4m3fn,flydsl,1100,0,4.4972,flydsl_bpreshuffle_wmma_t128x128x128_mw2_nw2_nb4_sk1_cm1_cn1,955.03,2331.62,0.0
gfx1250,256,4096,2048,512,torch.float8_e4m3fn,flydsl,1356,0,5.9174,flydsl_bpreshuffle_wmma_t256x128x128_mw2_nw2_nb4_sk1_cm1_cn1,1451.64,3366.84,0.0
gfx1250,256,8192,2048,512,torch.float8_e4m3fn,flydsl,1416,0,8.4972,flydsl_bpreshuffle_wmma_t256x256x256_mw2_nw2_nb2_sk1_cm1_cn1,2021.83,4565.89,0.0
gfx1250,256,16384,2048,512,torch.float8_e4m3fn,flydsl,1416,0,15.1011,flydsl_bpreshuffle_wmma_t256x256x256_mw2_nw2_nb2_sk1_cm1_cn1,2275.31,5068.91,0.0
gfx1250,256,32768,2048,512,torch.float8_e4m3fn,flydsl,1416,0,28.0945,flydsl_bpreshuffle_wmma_t256x256x256_mw2_nw2_nb2_sk1_cm1_cn1,2446.01,5411.86,0.0
Original file line number Diff line number Diff line change
Expand Up @@ -575,3 +575,28 @@ M,N,K,q_dtype_w
2048,6144,7168,torch.float8_e4m3fn
4096,6144,7168,torch.float8_e4m3fn
8192,6144,7168,torch.float8_e4m3fn
256,2112,7168,torch.float8_e4m3fn
512,2112,7168,torch.float8_e4m3fn
1024,2112,7168,torch.float8_e4m3fn
2048,2112,7168,torch.float8_e4m3fn
4096,2112,7168,torch.float8_e4m3fn
8192,2112,7168,torch.float8_e4m3fn
16384,2112,7168,torch.float8_e4m3fn
32768,2112,7168,torch.float8_e4m3fn
1,12288,1536,torch.float8_e4m3fn
2,12288,1536,torch.float8_e4m3fn
4,12288,1536,torch.float8_e4m3fn
8,12288,1536,torch.float8_e4m3fn
16,12288,1536,torch.float8_e4m3fn
32,12288,1536,torch.float8_e4m3fn
48,12288,1536,torch.float8_e4m3fn
64,12288,1536,torch.float8_e4m3fn
128,12288,1536,torch.float8_e4m3fn
256,12288,1536,torch.float8_e4m3fn
512,12288,1536,torch.float8_e4m3fn
1024,12288,1536,torch.float8_e4m3fn
2048,12288,1536,torch.float8_e4m3fn
4096,12288,1536,torch.float8_e4m3fn
8192,12288,1536,torch.float8_e4m3fn
16384,12288,1536,torch.float8_e4m3fn
32768,12288,1536,torch.float8_e4m3fn
Loading
Loading