diff --git a/torch_npu/csrc/core/npu/NPUEvent.cpp b/torch_npu/csrc/core/npu/NPUEvent.cpp index d956195154776cc6c972a55c9da23037be508959..38ff6321964683f12d3bf41e796f1ad54dc7ac38 100644 --- a/torch_npu/csrc/core/npu/NPUEvent.cpp +++ b/torch_npu/csrc/core/npu/NPUEvent.cpp @@ -107,7 +107,7 @@ float NPUEvent::elapsed_time(const NPUEvent& other) const "Both events must be recorded before calculating elapsed time.", PTA_ERROR(ErrCode::INTERNAL)); float time_ms = 0; NPUStatus ret = c10_npu::emptyAllNPUStream(); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("Failed to empty NPU task queue, ret: %s", ret.c_str()); } NPU_CHECK_ERROR_WITHOUT_UCE(aclrtSynchronizeEvent(event_)); @@ -130,7 +130,7 @@ uint64_t NPUEvent::recorded_time() const { TORCH_CHECK(is_created_, "Event must be recorded before getting recorded timestamp.", PTA_ERROR(ErrCode::INTERNAL)); NPUStatus ret = c10_npu::emptyAllNPUStream(); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("Failed to empty NPU task queue, ret: %s", ret.c_str()); } NPU_CHECK_ERROR_WITHOUT_UCE(aclrtSynchronizeEvent(event_)); @@ -151,7 +151,7 @@ void NPUEvent::synchronize() const { if (is_created_) { NPUStatus ret = c10_npu::emptyAllNPUStream(); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("MakeSureQueueEmpty fail, ret: %s", ret.c_str()); } NPU_CHECK_ERROR_WITHOUT_UCE(aclrtSynchronizeEvent(event_)); diff --git a/torch_npu/csrc/core/npu/NPUQueue.cpp b/torch_npu/csrc/core/npu/NPUQueue.cpp index fb234398da8759d39ef9839fbb6dc87470b07558..5605f9a4d39af33a2ba06f39517cbe899da02965 100644 --- a/torch_npu/csrc/core/npu/NPUQueue.cpp +++ b/torch_npu/csrc/core/npu/NPUQueue.cpp @@ -244,7 +244,7 @@ NPUStatus Repository::MakeSureQueueEmpty(bool check_error) std::string runtime_error; if (initialized == false) { ASCEND_LOGE("Task queue is not initialized, shouldn't call MakeSureQueueEmpty(). !!"); - return FAILED; + return NPU_STATUS_FAILED; } ASCEND_LOGI("Begin to makesure taskqueue empty."); // While waiting for ACL thread to launch tasks, @@ -348,7 +348,7 @@ NPUStatus Repository::MakeSureQueueEmpty(bool check_error) logger->debug("MakeSureQueueEmpty: clearing successful, device = %d, write_idx = %u, read_idx = %u, status = %d", device_idx, write_idx.idx, read_idx.idx, GetStatus()); - return SUCCESS; + return NPU_STATUS_SUCCESS; } bool Repository::WriteQueue(void *cur_paras) diff --git a/torch_npu/csrc/core/npu/NPUStream.cpp b/torch_npu/csrc/core/npu/NPUStream.cpp index e7d0e405bfd0fa7722d4fd1f602ba2884e28e685..9f85bff8152741b73fbb1b274d0afd651908feb5 100644 --- a/torch_npu/csrc/core/npu/NPUStream.cpp +++ b/torch_npu/csrc/core/npu/NPUStream.cpp @@ -339,7 +339,7 @@ aclrtStream NPUStream::stream() const AT_ASSERT(ptr, PTA_ERROR(ErrCode::PTR)); if (!this->isSyncLaunchStream() && ptr->repo->CheckInit()) { NPUStatus ret = ptr->repo->MakeSureQueueEmpty(); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("MakeSureQueueEmpty fail, ret: %s", ret.c_str()); return nullptr; } @@ -436,12 +436,12 @@ NPUStatus emptyAllNPUStream(bool check_error) } if (default_streamsi.stream != nullptr && default_streamsi.repo->CheckInit()) { ret = default_streamsi.repo->MakeSureQueueEmpty(check_error); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { return ret; } } } - return SUCCESS; + return NPU_STATUS_SUCCESS; } std::string getRepoInfo() @@ -488,7 +488,7 @@ bool npuSynchronizeDevice(bool check_error) { if (c10_npu::option::OptionsManager::GetTaskQueueEnable()) { NPUStatus ret = c10_npu::emptyAllNPUStream(check_error); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("MakeSureQueueEmpty fail, ret: %s", ret.c_str()); } } @@ -516,7 +516,7 @@ bool npuSynchronizeUsedDevices(bool check_error) { if (c10_npu::option::OptionsManager::GetTaskQueueEnable()) { NPUStatus ret = c10_npu::emptyAllNPUStream(check_error); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("MakeSureQueueEmpty fail, ret: %s", ret.c_str()); } } diff --git a/torch_npu/csrc/core/npu/npu_log.h b/torch_npu/csrc/core/npu/npu_log.h index 9cd3e5ec55422cfeb5de220b00c30d0932cf298a..f511f7cb910bb8ce8433d114ef4fc54f5b10e335 100644 --- a/torch_npu/csrc/core/npu/npu_log.h +++ b/torch_npu/csrc/core/npu/npu_log.h @@ -7,11 +7,11 @@ #include "torch_npu/csrc/core/npu/register/OptionsManager.h" #define NPUStatus std::string -#define SUCCESS "SUCCESS" +#define NPU_STATUS_SUCCESS "SUCCESS" #define INTERNEL_ERROR "INTERNEL_ERROR" #define PARAM_ERROR "PARAM_ERROR" #define ALLOC_ERROR "ALLOC_ERROR" -#define FAILED "FAILED" +#define NPU_STATUS_FAILED "FAILED" #define ASCEND_LOGE(fmt, ...) \ do { \ diff --git a/torch_npu/csrc/framework/NPUDefine.cpp b/torch_npu/csrc/framework/NPUDefine.cpp index 2bf90fffaea9e0f502fab76707bab66632b198b9..5902ceb9f0f02cb08a0aa0fb4af7157d3b3c3a16 100644 --- a/torch_npu/csrc/framework/NPUDefine.cpp +++ b/torch_npu/csrc/framework/NPUDefine.cpp @@ -12,7 +12,7 @@ void ExecuteParas::Release() } DestroyConstParams(constParams); NPUStatus ret = DestroyAclParams(paras); - if (ret != SUCCESS) { + if (ret != NPU_STATUS_SUCCESS) { ASCEND_LOGE("DestroyAclParams fail, ret: %s", ret.c_str()); } hostMemory.clear(); @@ -92,7 +92,7 @@ void ExecuteParas::Release() params.input_data_buf = nullptr; params.output_desc = nullptr; params.output_data_buf = nullptr; - return SUCCESS; + return NPU_STATUS_SUCCESS; } void DestroyConstParams(CONST_PARAMS ¶ms) diff --git a/torch_npu/csrc/framework/utils/CalcuOpUtil.cpp b/torch_npu/csrc/framework/utils/CalcuOpUtil.cpp index 4400c3e7579fc1decd5824fdda4cef02cfe3c56d..8339bd50228f128428eac019631ed871585af41a 100644 --- a/torch_npu/csrc/framework/utils/CalcuOpUtil.cpp +++ b/torch_npu/csrc/framework/utils/CalcuOpUtil.cpp @@ -191,7 +191,7 @@ NPUStatus CalcuOpUtil::AclrtMemcpyAsync(const std::pair &ds NPU_CHECK_ERROR( c10_npu::queue::LaunchAsyncCopyTask(dst_ptr, dst_size, const_cast(src_ptr), src_size, kind)); - return "SUCCESS"; + return "NPU_STATUS_SUCCESS"; } aclError CalcuOpUtil::AclrtMemcpyWithModeSwitch(const StorageAndOffsetMemSizePair &dst, diff --git a/torch_npu/csrc/npu/Module.cpp b/torch_npu/csrc/npu/Module.cpp index f187a0dd8c75db57fc2b3c5457f7726fd233ddc7..65da2f7d4c37c7f8a86b550cc4e1150619104b89 100644 --- a/torch_npu/csrc/npu/Module.cpp +++ b/torch_npu/csrc/npu/Module.cpp @@ -1621,7 +1621,7 @@ PyObject* THNPModule_aclnn_reselect_static_kernel(PyObject* self, PyObject* noar { HANDLE_TH_ERRORS NPUStatus ret = c10_npu::emptyAllNPUStream(); - TORCH_CHECK(ret == SUCCESS, "Failed to empty NPU task queue, ret:", ret, PTA_ERROR(ErrCode::INTERNAL)); + TORCH_CHECK(ret == NPU_STATUS_SUCCESS, "Failed to empty NPU task queue, ret:", ret, PTA_ERROR(ErrCode::INTERNAL)); static const auto task_queue_enable = c10_npu::option::OptionsManager::GetTaskQueueEnable(); if (task_queue_enable == 2) { @@ -1631,7 +1631,7 @@ PyObject* THNPModule_aclnn_reselect_static_kernel(PyObject* self, PyObject* noar }; at_npu::native::OpCommand::RunOpApiV2("reselect_static_kernel", acl_call); NPUStatus ret = c10_npu::emptyAllNPUStream(); - TORCH_CHECK(ret == SUCCESS, "Failed to empty NPU task queue, ret:", ret, PTA_ERROR(ErrCode::INTERNAL)); + TORCH_CHECK(ret == NPU_STATUS_SUCCESS, "Failed to empty NPU task queue, ret:", ret, PTA_ERROR(ErrCode::INTERNAL)); } else { c10_npu::opapi::ReselectStaticKernel(); }