From 12475b65e9f513b2e0028ff8a2e7672ee5c41d47 Mon Sep 17 00:00:00 2001 From: dongwenbo6 Date: Mon, 19 Feb 2024 16:51:44 +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 ec26c9a952..87c5b72756 100644 --- a/torch_npu/csrc/core/npu/NPUQueue.cpp +++ b/torch_npu/csrc/core/npu/NPUQueue.cpp @@ -216,11 +216,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); } #ifndef BUILD_LIBTORCH @@ -292,11 +292,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