TensorRT-LLMs/cpp/tensorrt_llm/plugins/gemmAllReducePlugin
xavier-nvidia 200ea9ee81
fix TMA error with GEMM+AR on TP=2 (#6075)
Signed-off-by: Xavier Simmons <xsimmons@nvidia.com>
2025-07-18 10:26:08 +08:00
..
CMakeLists.txt Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
gemmAllReducePlugin.cpp fix TMA error with GEMM+AR on TP=2 (#6075) 2025-07-18 10:26:08 +08:00
gemmAllReducePlugin.h fix TMA error with GEMM+AR on TP=2 (#6075) 2025-07-18 10:26:08 +08:00
gemmAllReducePluginProfiler.cpp fix TMA error with GEMM+AR on TP=2 (#6075) 2025-07-18 10:26:08 +08:00
gemmAllReducePluginProfiler.h Fix GEMM+AR fusion on blackwell (#5563) 2025-07-09 08:48:47 +08:00
gemmAllReducePluginResource.h refactoring: port customized kernels with public cutlass version (#5027) 2025-06-13 16:19:31 +08:00