From dff1d36b6361490215d4892d9a0c7409a65c3b99 Mon Sep 17 00:00:00 2001 From: dongwenbo6 Date: Mon, 19 Feb 2024 19:10:41 +0800 Subject: [PATCH] fix broken pipeline --- torch_npu/csrc/core/npu/NPUQueue.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_npu/csrc/core/npu/NPUQueue.cpp b/torch_npu/csrc/core/npu/NPUQueue.cpp index 02946c467f..c39f88653f 100644 --- a/torch_npu/csrc/core/npu/NPUQueue.cpp +++ b/torch_npu/csrc/core/npu/NPUQueue.cpp @@ -209,11 +209,11 @@ NPUStatus Repository::MakeSureQueueEmpty() { if (GetStatus() == RepoStatus::ERROR_EXIT) { // Avoid repeatedly throwing exceptions - SetStatus(CAN_EXIT); throw std::runtime_error("The Inner error is reported as above.\n "\ "Since the operator is called asynchronously, the stacktrace may be inaccurate. "\ "If you want to get the accurate stacktrace, "\ "pleace set the environment variable ASCEND_LAUNCH_BLOCKING=1."); + SetStatus(CAN_EXIT); } // Get the GIL @@ -280,11 +280,11 @@ void Repository::Enqueue(void* cur_paras) { if (GetStatus() == RepoStatus::ERROR_EXIT) { // Avoid repeatedly throwing exceptions - SetStatus(CAN_EXIT); throw std::runtime_error("The Inner error is reported as above.\n "\ "Since the operator is called asynchronously, the stacktrace may be inaccurate. "\ "If you want to get the accurate stacktrace, "\ "pleace set the environment variable ASCEND_LAUNCH_BLOCKING=1."); + SetStatus(CAN_EXIT); } if (GetStatus() != RUN && GetStatus() != INIT) { -- Gitee