Skip to content

Commit be4181a

Browse files
committed
fix hint manager import error
1 parent 1ffece8 commit be4181a

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

python/triton/compiler/hint_manager.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,21 +42,21 @@ def __init__(self, backend_name):
4242
def _load_handler(self, backend):
4343
if backend == 'npu':
4444
try:
45-
module = importlib.import_module("third_party.ascend.backend.ascend_hint_handler")
45+
module = importlib.import_module("triton.backends.ascend.ascend_hint_handler")
4646
return module.AscendHintHandler()
4747
except ImportError as e:
4848
print(f"[FlagTree] Warning: Failed to load Ascend Hint Handler: {e}", file=sys.stderr)
4949
return BaseHintHandler()
5050
elif backend == 'aipu':
5151
try:
52-
module = importlib.import_module("third_party.aipu.backend.aipu_hint_handler")
52+
module = importlib.import_module("triton.backends.aipu.aipu_hint_handler")
5353
return module.AipuHintHandler()
5454
except ImportError as e:
5555
print(f"[FlagTree] Warning: Failed to load aipu Hint Handler: {e}", file=sys.stderr)
5656
return BaseHintHandler()
5757
elif backend == 'cuda':
5858
try:
59-
module = importlib.import_module("third_party.nvidia.backend.nvidia_hint_handler")
59+
module = importlib.import_module("triton.backends.nvidia.nvidia_hint_handler")
6060
return module.NvidiaHintHandler()
6161
except ImportError as e:
6262
print(f"[FlagTree] Warning: Failed to load Nvidia Hint Handler: {e}", file=sys.stderr)

0 commit comments

Comments
 (0)