Yukun He a1645c922b Fallback to NCCL for various patterns when input size is large. (#4009) ... When input size is larger than the max workspace size, we shall fallback to NCCL + corresponding pre/post function to ensure the functionality of AllReduce. Signed-off-by: Yukun He <23156053+hyukn@users.noreply.github.com>		2025-05-01 15:17:16 -07:00
..
test_allreduce.py	Fallback to NCCL for various patterns when input size is large. (#4009)	2025-05-01 15:17:16 -07:00
test_ar_residual_norm.py	refactor: remove ParallelConfig in tensorrt_llm._torch.distributed module (#3370)	2025-04-11 15:34:20 -07:00
test_deepseek_allreduce.py	Clean up allreduce op in Deepseek V3 model. (#3829)	2025-05-01 07:56:36 +08:00
test_embedding.py	refactor: remove ParallelConfig in tensorrt_llm._torch.distributed module (#3370)	2025-04-11 15:34:20 -07:00
test_linear.py	refactor: remove ParallelConfig in tensorrt_llm._torch.distributed module (#3370)	2025-04-11 15:34:20 -07:00
test_star_attention_input.jsonl	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
test_star_attention.py	feat: Support cos_sin_cache in all cases. (#3517)	2025-04-16 13:48:44 +08:00
test_user_buffers.py	fix: Remove ParallelConfig. (#3678)	2025-04-21 14:14:08 +08:00