File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments