Merge pull request #2971 from rsmallblue/tvm-ffi

[CuTeDSL]fix tvm-ffi path in from_dlpack
2026-02-14 14:14:10 +08:00
parent d4bbf728ca 395ab575f6
commit ec7e6cb17b
1 changed files with 1 additions and 1 deletions
--- 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: