diff --git a/python/CuTeDSL/cutlass/cute/runtime.py b/python/CuTeDSL/cutlass/cute/runtime.py index d4f8c554..4019f1b6 100644 --- a/python/CuTeDSL/cutlass/cute/runtime.py +++ b/python/CuTeDSL/cutlass/cute/runtime.py @@ -144,7 +144,7 @@ class _Tensor(Tensor): elif enable_tvm_ffi: import tvm_ffi - self._tvm_ffi_tensor = tvm_ffi.from_dlpack(tensor) + self._tvm_ffi_tensor = tvm_ffi.from_dlpack(tensor, stream=-1) self._dlpack_data = self._tvm_ffi_tensor.__dlpack__() else: try: