TensorRT-LLM
TensorRT-LLM copied to clipboard
feat: Unify two versions of allreduce custom op
Unify the two versions of AllReduce op in Module and custome op levels.