kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-28 06:33:15 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
cbca6505ff
TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/batchedGemm/cubin
History
Olya Kozlova b3e6723dbc
feat: Adding FP8 BMM from Codegen (#3541) 
* Adding FP8 BMM from Codegen

Signed-off-by: Olya Kozlova <okozlova@s4124-0110.nvidia.com>

* Fixed licenses

Signed-off-by: Olya Kozlova <okozlova@s4124-0062.nvidia.com>

---------

Signed-off-by: Olya Kozlova <okozlova@s4124-0110.nvidia.com>
Signed-off-by: Olya Kozlova <okozlova@s4124-0062.nvidia.com>
Co-authored-by: Olya Kozlova <okozlova@6u1g-0018.nvidia.com>
Co-authored-by: Olya Kozlova <okozlova@s4124-0062.nvidia.com>
2025-04-16 10:37:15 +02:00
..
BatchedGemmKernel_BatchM_E4m3Fp32_Bfloat16_Tile128x128x128_EpiTile128x64_Mma128x64x32_Cluster1x1x1_DsFp8_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchM_E4m3Fp32_Bfloat16_Tile128x128x128_EpiTile128x128_Mma128x128x32_Cluster1x1x1_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchM_E4m3Fp32_E4m3_Tile128x128x128_EpiTile128x64_Mma128x64x32_Cluster1x1x1_DsFp8_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchM_E4m3Fp32_E4m3_Tile128x128x128_EpiTile128x128_Mma128x128x32_Cluster1x1x1_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchN_E4m3Fp32_Bfloat16_Tile128x128x128_EpiTile64x128_Mma64x128x32_Cluster1x1x1_transposeMmaOutput_DsFp8_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchN_E4m3Fp32_Bfloat16_Tile128x128x128_EpiTile128x128_Mma128x128x32_Cluster1x1x1_transposeMmaOutput_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchN_E4m3Fp32_E4m3_Tile128x128x128_EpiTile64x128_Mma64x128x32_Cluster1x1x1_transposeMmaOutput_DsFp8_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00
BatchedGemmKernel_BatchN_E4m3Fp32_E4m3_Tile128x128x128_EpiTile128x128_Mma128x128x32_Cluster1x1x1_transposeMmaOutput_sm100a_cubin.h feat: Adding FP8 BMM from Codegen (#3541) 2025-04-16 10:37:15 +02:00