NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-04-19 14:28:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

pass in stream=-1

Browse Source
This commit is contained in:
yuanxiaolan
2026-02-03 11:55:54 +08:00
parent de198b2419
commit de161925a5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/CuTeDSL/cutlass/cute/runtime.py
View File

@@ -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: