diff --git a/torch_npu/csrc/core/npu/NPUQueue.cpp b/torch_npu/csrc/core/npu/NPUQueue.cpp index 4c5ab672a296896e1177a9b2afa9ee0d386fc9d2..e774c281b58ee84f10f2860064fd828280ba4ffc 100644 --- a/torch_npu/csrc/core/npu/NPUQueue.cpp +++ b/torch_npu/csrc/core/npu/NPUQueue.cpp @@ -3,6 +3,7 @@ #include "torch_npu/csrc/core/npu/npu_log.h" #include "torch_npu/csrc/framework/utils/NpuUtils.h" #include "torch_npu/csrc/core/npu/NPUFunctions.h" +#include "torch_npu/csrc/framework/OpParamMaker.h" #include #include @@ -288,8 +289,20 @@ void Repository::Enqueue(void* cur_paras) { } if (GetStatus() != RUN && GetStatus() != INIT) { - ASCEND_LOGE("Task queue thread is exit, cann't call Enqueue(). !!"); - return; + auto queueParam = static_cast(cur_paras); + auto type = queueParam->paramType; + if (type ==c10_npu::queue::COMPILE_AND_EXECUTE) { + auto cur_paras = static_cast(queueParam->paramVal); + auto op_param = cur_paras->opType; + ASCEND_LOGE("Task queue thread is exit, cann't call Enqueue(). op name is=%s", *op_name); + } else if (type ==c10_npu::queue::ASYNC_MEMCPY) { + auto cur_paras = static_cast(queueParam->paramVal); + ASCEND_LOGE("Task queue thread is exit, cann't call Enqueue(). src=%p, dst=%p", cur_paras->src, cur_paras->dst); + } else { + auto cur_paras = static_cast(queueParam->paramVal); + ASCEND_LOGE("Task queue thread is exit, cann't call Enqueue(). event is=%p", cur_paras->event); + } + return; } bool ret = false; ssize_t s;