mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
* [feat] integrate ada blockwise gemm Signed-off-by: CarstyYou <xiy@nvidia.com> * [fix] align scale M Signed-off-by: CarstyYou <xiy@nvidia.com> * [feat] swizzle mma output Signed-off-by: CarstyYou <xiy@nvidia.com> * [test] add ut for sm89 Signed-off-by: CarstyYou <xiy@nvidia.com> * [delete] remove useless comments Signed-off-by: CarstyYou <xiy@nvidia.com> * [chore] codestyle Signed-off-by: CarstyYou <xiy@nvidia.com> * [fix] fix review comments Signed-off-by: CarstyYou <xiy@nvidia.com> * [chore] fix license Signed-off-by: CarstyYou <xiy@nvidia.com> * [chore] fix license Signed-off-by: CarstyYou <xiy@nvidia.com> --------- Signed-off-by: CarstyYou <xiy@nvidia.com> Co-authored-by: bhsueh_NV <11360707+byshiue@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| allreduce_gemm | ||
| fp8_blockscale_gemm | ||
| fp8_rowwise_gemm | ||
| fpA_intB_gemm | ||
| fused_gated_gemm | ||
| int8_gemm | ||
| python | ||
| CMakeLists.txt | ||
| cutlass_heuristic.cpp | ||
| cutlass_heuristic.h | ||
| cutlass_preprocessors.cpp | ||
| cutlass_preprocessors.h | ||
| cutlass_type_conversion.h | ||