[Contrib] Update RandomFill to use StreamSync for CUDA synchronization (#18469)

guan404ming · web-flow · commit 12f3bb0c6f5a · 2025-11-20T13:23:45.000+09:00
diff --git 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() {

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ void RandomFill(DLTensor* tensor) {`
`110`	`110`	`} else {`
`111`	`111`	`LOG(FATAL) << "ValueError: Unsupported dtype: " << tensor->dtype;`
`112`	`112`	`}`
`113`		`- TVMSynchronize(tensor->device.device_type, tensor->device.device_type, nullptr);`
	`113`	`+ cuda_api->StreamSync(tensor->device, nullptr);`
`114`	`114`	`}`
`115`	`115`
`116`	`116`	`TVM_FFI_STATIC_INIT_BLOCK() {`