[Bugfix] [NPU] Fix check_env on Ascend for CANN 8.5 (#22888)

2026-04-17 04:05:20 +03:00
parent 3d2d57c6cc
commit ba850d3a9d
1 changed files with 4 additions and 1 deletions
@@ -362,7 +362,10 @@ class NPUEnv(BaseEnv):
        else:
            cann_info["CANN"] = "Not Available"
        try:
-            bisheng = os.path.join(CANN_HOME, "compiler/ccec_compiler/bin/bisheng")
+            bisheng = os.path.join(CANN_HOME, "tools/bisheng_compiler/bin/bisheng")
+            if not os.path.isfile(bisheng):
+                # Check path for old CANN version
+                bisheng = os.path.join(CANN_HOME, "compiler/ccec_compiler/bin/bisheng")
            bisheng_output = (
                subprocess.check_output([bisheng, "--version"]).decode("utf-8").strip()
            )