From c866321f3b21f3d6a3b7c480df3d7b4c11852a9c Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Wed, 10 Sep 2025 15:28:45 -0400 Subject: [PATCH] [Python] Fix runtime tensor import This PR fixes a few places where the python import of runtime tensor is incorrect. The error wasn't revealed in the previous NDArray->Tensor rename PR since these imports are not at the top level. --- python/tvm/exec/disco_worker.py | 2 +- python/tvm/testing/runner.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/tvm/exec/disco_worker.py b/python/tvm/exec/disco_worker.py index 5b20480decd4..7fe94c6cb0df 100644 --- a/python/tvm/exec/disco_worker.py +++ b/python/tvm/exec/disco_worker.py @@ -24,7 +24,7 @@ import tvm from tvm_ffi import get_global_func, register_global_func from tvm.runtime import Tensor, ShapeTuple, String -from tvm.runtime.tensor import tensor +from tvm.runtime import tensor @register_global_func("tests.disco.add_one", override=True) diff --git a/python/tvm/testing/runner.py b/python/tvm/testing/runner.py index f2625b28f972..be50cc8707c5 100644 --- a/python/tvm/testing/runner.py +++ b/python/tvm/testing/runner.py @@ -32,7 +32,7 @@ def _args_to_device(args, device): import numpy as np - from tvm.runtime.tensor import Tensor, empty + from tvm.runtime import Tensor, empty uploaded_args = [] for arg in args: @@ -46,7 +46,7 @@ def _args_to_device(args, device): def _args_to_numpy(args): - from tvm.runtime.tensor import Tensor + from tvm.runtime import Tensor downloaded_args = [] for arg in args: