[tvm] 01/01: [Fix] Fix Build with RCCL

[ju...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

junrushao pushed a commit to branch feature/2023-10-03/fix-rccl-build
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit 6e0e4f47d5979f4d0f26f3e2b0c9bb6c0a341448
Author: Junru Shao <ju...@gmail.com>
AuthorDate: Tue Oct 3 12:18:26 2023 -0400

    [Fix] Fix Build with RCCL
    
    This PR contains a minor fix for RCCL integration.
---
 src/runtime/disco/nccl/nccl.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/runtime/disco/nccl/nccl.cc b/src/runtime/disco/nccl/nccl.cc
index 0ce8b985ff..f9d3ca400a 100644
--- a/src/runtime/disco/nccl/nccl.cc
+++ b/src/runtime/disco/nccl/nccl.cc
@@ -196,7 +196,7 @@ void AllGather(NDArray send, NDArray recv) {
   CCLThreadLocalContext* ctx = CCLThreadLocalContext::Get();
   ShapeTuple shape = send.Shape();
   int64_t numel = shape->Product();
-  cudaStream_t stream = ctx->GetDefaultStream();
+  deviceStream_t stream = ctx->GetDefaultStream();
   NCCL_CALL(ncclAllGather(send->data, recv->data, numel,
                           /*datatype=*/AsNCCLDataType(DataType(send->dtype)), ctx->comm, stream));
 }