diff --git a/torch_npu/csrc/core/npu/interface/AclInterface.cpp b/torch_npu/csrc/core/npu/interface/AclInterface.cpp
index 90e6f8ca82126846a69cbcf8d2b1eded42f0618e..1ca820dba6b502f96f531659f3022d2d47bf89d9 100644
--- a/torch_npu/csrc/core/npu/interface/AclInterface.cpp
+++ b/torch_npu/csrc/core/npu/interface/AclInterface.cpp
@@ -1126,14 +1126,12 @@ aclError AclrtUnSubscribeReport(uint64_t theadId, aclrtStream stream)
 bool AclrtMemcpyAsyncWithConditionExist()
 {
     const static bool isAclrtMemcpyAsyncWithConditionExist = []() -> bool {
-        auto func = GET_FUNC(aclrtMemcpyAsyncWithCondition)
-        bool flag = c10_npu::GetSocVersion() >= c10_npu::SocVersion::Ascend910B1 ? true : false;
+        auto func = GET_FUNC(aclrtMemcpyAsyncWithCondition);
         if (func != nullptr) {
             ASCEND_LOGI("Successfully to find function aclrtMemcpyAsyncWithCondition");
-        } else {
-            flag = false;
+            return c10_npu::GetSocVersion() >= c10_npu::SocVersion::Ascend910B1;
         }
-        return flag;
+        return false;
     }();
     return isAclrtMemcpyAsyncWithConditionExist;
 }