update hintmanager, Align with triton_v3.5.x branch.

Author: starrryz

python/triton/compiler/hint_manager.py

@@ -1,4 +1,3 @@
-import os
 import sys
 import importlib
 
@@ -49,24 +48,32 @@ def _load_handler(self, backend):
                 print(f"[FlagTree] Warning: Failed to load Ascend Hint Handler: {e}", file=sys.stderr)
                 return BaseHintHandler()
         elif backend == 'aipu':
-            from .backends.aipu import AipuHintHandler
-            return AipuHintHandler()
+            try:
+                module = importlib.import_module("third_party.aipu.backend.aipu_hint_handler")
+                return module.AipuHintHandler()
+            except ImportError as e:
+                print(f"[FlagTree] Warning: Failed to load aipu Hint Handler: {e}", file=sys.stderr)
+                return BaseHintHandler()
+        elif backend == 'cuda':
+            try:
+                module = importlib.import_module("third_party.nvidia.backend.nvidia_hint_handler")
+                return module.NvidiaHintHandler()
+            except ImportError as e:
+                print(f"[FlagTree] Warning: Failed to load Nvidia Hint Handler: {e}", file=sys.stderr)
+                return BaseHintHandler()
         else:
             return BaseHintHandler()
 
 
 # supported backend with matched version
-SUPPORTED_CONFIG = {
-    "cuda": {"3.5"},
-    "npu": {"3.2"},
-    "aipu": {"3.3"},
-}
+SUPPORTED_BACKENDS = ["aipu", "npu", "cuda"]
 
+# TODO : npu will have conflicts if more backend involved
 # mapping name
 BACKEND_ALIASES = {
     "ascend": "npu",
     "huawei": "npu",
-    "nv": "cuda",
+    "nvidia": "cuda",
 }
 
 
@@ -81,7 +88,6 @@ def hint_get_flagtree_backend() -> str:
     detected_backend = ""
 
     import torch
-    import triton
 
     # Priority 1: Triton Driver
     try:
@@ -96,42 +102,24 @@ def hint_get_flagtree_backend() -> str:
     except ImportError:
         pass
 
+    # TODO : some backend may not support priority 1, so keep priority 2 is necessary
     # Priority 2: Torch Global State
     if not detected_backend:
-        candidates = list(SUPPORTED_CONFIG.keys())
-        # cuda priority least
-        candidates.sort(key=lambda x: 1 if x == "cuda" else 0)
+        check_priority = ["aipu", "npu", "cuda"]
 
         # 3. parse according to benefit
-        for candidate in candidates:
-            module_name = candidate
-            module = getattr(torch, module_name, None)
+        for candidate in check_priority:
+            module = getattr(torch, candidate, None)
             if module and hasattr(module, "is_available") and module.is_available():
                 detected_backend = candidate
                 break
 
-    # Priority 3: Environment Variable (need to remove!!!)
-    if not detected_backend:
-        detected_backend = os.environ.get("FLAGTREE_BACKEND", "")
-
     # (Normalization and Validation)
     canonical_backend = normalize_backend_name(detected_backend)
 
-    if not canonical_backend or canonical_backend not in SUPPORTED_CONFIG:
+    if not canonical_backend or canonical_backend not in SUPPORTED_BACKENDS:
         return ""
 
-    # verify name and version match
-    try:
-        current_triton_version = ".".join(triton.__version__.split(".")[:2])
-        supported_versions = SUPPORTED_CONFIG[canonical_backend]
-        if current_triton_version not in supported_versions:
-            msg = (f"[Flagtree] Hint ignored: Detected backend '{canonical_backend}' but current Triton version "
-                   f"'{current_triton_version}' matches no supported versions {supported_versions}.")
-            print(msg, file=sys.stderr)
-            return ""
-    except Exception:
-        pass
-
     return canonical_backend