TensorRT-LLMs/cpp/tensorrt_llm/plugins/gemmAllReducePlugin
xavier-nvidia 7b818de700
Fix TMA error with GEMM+AR on TP=2 (#6071)
Signed-off-by: Xavier Simmons <xsimmons@nvidia.com>
2025-07-16 10:27:32 -07: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 (#6071) 2025-07-16 10:27:32 -07:00
gemmAllReducePlugin.h Fix TMA error with GEMM+AR on TP=2 (#6071) 2025-07-16 10:27:32 -07:00
gemmAllReducePluginProfiler.cpp Fix TMA error with GEMM+AR on TP=2 (#6071) 2025-07-16 10:27:32 -07:00
gemmAllReducePluginProfiler.h Fix GEMM+AR nvbugs 5219533,5127801,5072306 (#5969) 2025-07-11 10:22:02 -07:00
gemmAllReducePluginResource.h refactoring: port customized kernels with public cutlass version (#5027) 2025-06-13 16:19:31 +08:00