From 2835adf858f6001fe15e207030f6e7afab3b9cc4 Mon Sep 17 00:00:00 2001 From: dongwenbo6 Date: Sat, 2 Mar 2024 14:28:13 +0800 Subject: [PATCH] bug fix --- torch_npu/csrc/npu/Module.cpp | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/torch_npu/csrc/npu/Module.cpp b/torch_npu/csrc/npu/Module.cpp index d421c5a1d5..f5cb2d19b3 100644 --- a/torch_npu/csrc/npu/Module.cpp +++ b/torch_npu/csrc/npu/Module.cpp @@ -294,13 +294,7 @@ PyObject* THNPModule_setDevice_wrap(PyObject* self, PyObject* arg) } } - int pre_device = 0; - auto ret = c10_npu::GetDevice(&pre_device); - if (ret != ACL_ERROR_NONE) { - NPU_CHECK_ERROR(c10_npu::SetDevice(device)); - } else if (pre_device != device) { - c10_npu::NpuSysCtrl::GetInstance().ExchangeDevice(pre_device, device); - } + NPU_CHECK_ERROR(c10_npu::SetDevice(device)); Py_RETURN_NONE; END_HANDLE_TH_ERRORS -- Gitee