diff --git a/torch_npu/csrc/core/npu/NPUQueue.cpp b/torch_npu/csrc/core/npu/NPUQueue.cpp index b91a5fc507a28acf184231c5ebc93b5e7ad4a0e6..95efe6e9e01a3d4f84394d0277ca884b498cfa53 100644 --- a/torch_npu/csrc/core/npu/NPUQueue.cpp +++ b/torch_npu/csrc/core/npu/NPUQueue.cpp @@ -3,7 +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" #ifndef BUILD_LIBTORCH #include #endif @@ -300,7 +300,19 @@ void Repository::Enqueue(void* cur_paras) { } if (GetStatus() != RUN && GetStatus() != INIT) { - ASCEND_LOGE("Task queue thread is exit, cann't call Enqueue(). !!"); + 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;