From ccd80a88836bb1c25ece24f7a1b41dafd6ba87f8 Mon Sep 17 00:00:00 2001 From: lvyufeng Date: Sun, 28 Sep 2025 11:31:10 +0000 Subject: [PATCH] fix ms2.7 on GPU --- mindtorch/_apis/cpu.py | 5 +++-- mindtorch/_apis/gpu.py | 2 +- mindtorch/executor.py | 4 ++++ tests/{triton => test_triton}/__init__.py | 0 tests/{triton => test_triton}/test_add.py | 0 5 files changed, 8 insertions(+), 3 deletions(-) rename tests/{triton => test_triton}/__init__.py (100%) rename tests/{triton => test_triton}/test_add.py (100%) diff --git a/mindtorch/_apis/cpu.py b/mindtorch/_apis/cpu.py index 247f8c02d..1da15e7a7 100644 --- a/mindtorch/_apis/cpu.py +++ b/mindtorch/_apis/cpu.py @@ -3,12 +3,13 @@ import math import numpy as np import mindspore -from mindspore._c_expression import _empty_instance +from mindspore.ops.auto_generate.gen_ops_prim import Empty import mindtorch from .._op_prim.cpu import legacy +empty_op = Empty().set_device('CPU') def empty(size, dtype): - return _empty_instance(size, dtype=dtype, device='CPU') + return empty_op(size, dtype=dtype, device='CPU') def inplace_normal(input, mean, std, generator_): out = np.random.normal(mean, std, input.shape).astype(mindtorch.dtype2np[input.dtype]) diff --git a/mindtorch/_apis/gpu.py b/mindtorch/_apis/gpu.py index 8c11c4d8f..815795018 100644 --- a/mindtorch/_apis/gpu.py +++ b/mindtorch/_apis/gpu.py @@ -1,9 +1,9 @@ import numbers import math +from packaging import version import numpy as np import mindspore -from mindspore._c_expression import _empty_instance import mindtorch from .._op_prim.gpu import legacy diff --git a/mindtorch/executor.py b/mindtorch/executor.py index 991c98f6e..20176c7ca 100644 --- a/mindtorch/executor.py +++ b/mindtorch/executor.py @@ -1,8 +1,12 @@ import mindtorch from .dispatcher import dispatcher +from .configs import MS27 def execute(func_name, *args, **kwargs): out, device = dispatcher.dispatch(func_name, *args, **kwargs) + # if MS27: + # return out + if not isinstance(out, (tuple, list)): out._device = device else: diff --git a/tests/triton/__init__.py b/tests/test_triton/__init__.py similarity index 100% rename from tests/triton/__init__.py rename to tests/test_triton/__init__.py diff --git a/tests/triton/test_add.py b/tests/test_triton/test_add.py similarity index 100% rename from tests/triton/test_add.py rename to tests/test_triton/test_add.py