Skip to content

Commit 005c18d

Browse files
ilmarkovilmarkov
authored andcommitted
[Distributed] Add custom allreduce support for ROCM (vllm-project#14125)
Signed-off-by: ilmarkov <imarkov@redhat.com> Co-authored-by: ilmarkov <imarkov@redhat.com> Signed-off-by: Bill Nell <bnell@redhat.com>
1 parent 8a72a9c commit 005c18d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/custom_all_reduce.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -602,4 +602,4 @@ class CustomAllreduce {
602602
* template void vllm::CustomAllreduce::allreduce<half>(cudaStream_t, half *,
603603
half *, int, int, int);
604604
*/
605-
} // namespace vllm
605+
} // namespace vllm

0 commit comments

Comments
 (0)