This is an automated email from the ASF dual-hosted git repository.
mshr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 12f3bb0c6f [Contrib] Update RandomFill to use StreamSync for CUDA
synchronization (#18469)
12f3bb0c6f is described below
commit 12f3bb0c6f5abff27de657a7b9afc6031a767d35
Author: Guan-Ming (Wesley) Chiu <[email protected]>
AuthorDate: Thu Nov 20 12:23:45 2025 +0800
[Contrib] Update RandomFill to use StreamSync for CUDA synchronization
(#18469)
---
src/runtime/contrib/curand/curand.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/runtime/contrib/curand/curand.cc
b/src/runtime/contrib/curand/curand.cc
index 2a43d309e7..53505770f8 100644
--- a/src/runtime/contrib/curand/curand.cc
+++ b/src/runtime/contrib/curand/curand.cc
@@ -110,7 +110,7 @@ void RandomFill(DLTensor* tensor) {
} else {
LOG(FATAL) << "ValueError: Unsupported dtype: " << tensor->dtype;
}
- TVMSynchronize(tensor->device.device_type, tensor->device.device_type,
nullptr);
+ cuda_api->StreamSync(tensor->device, nullptr);
}
TVM_FFI_STATIC_INIT_BLOCK() {
