From de161925a50e7d96bb36ed549a09bf2769e64cc5 Mon Sep 17 00:00:00 2001 From: yuanxiaolan Date: Tue, 3 Feb 2026 11:55:54 +0800 Subject: [PATCH] pass in stream=-1 --- python/CuTeDSL/cutlass/cute/runtime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: