fix code format problems

Author: motinwing

python/triton/compiler/code_generator.py

@@ -16,6 +16,7 @@
 # ideally we wouldn't need any runtime component
 from ..runtime import JITFunction
 from .._utils import find_paths_if, get_iterable_path, set_iterable_path
+from .hint_manager import hint_trigger
 
 from .errors import (CompilationError, CompileTimeAssertionFailure, UnsupportedLanguageConstruct)
 

python/triton/runtime/jit.py

@@ -12,8 +12,6 @@
 from ..runtime.driver import driver
 from types import ModuleType
 from .._utils import find_paths_if, get_iterable_path
-import tokenize
-from io import StringIO
 
 TRITON_MODULE = __name__[:-len(".runtime.jit")]
 
@@ -705,6 +703,7 @@ def preload(self, specialization_data):
     # the user might want to monkey-patch self.src dynamically.
     # Our unit tests do this, for example.
     def parse(self):
+        from ..compiler.hint_manager import hint_trigger
         # Maps line numbers to comment hints
         line_flagtree_hints = hint_trigger("maps_line_numbers_to_comment_hints", self)
         if line_flagtree_hints is None:

third_party/aipu/backend/aipu_hint_handler.py

@@ -21,11 +21,11 @@ def get_node_hints(code_generator, node):
     @staticmethod
     def inject_kwargs_with_hints(fn, flagtree_hints, line_num, kws):
         if fn.__name__ == "load" and flagtree_hints is not None:
-                    print(f"[FLAGTREE] tl.load at line {line_num} has annotation {flagtree_hints}")
-                    if 'flagtree_hints' not in kws:
-                        kws['flagtree_hints'] = ""
-                    if flagtree_hints not in kws['flagtree_hints']:
-                        kws['flagtree_hints'] = flagtree_hints
+                print(f"[FLAGTREE] tl.load at line {line_num} has annotation {flagtree_hints}")
+                if 'flagtree_hints' not in kws:
+                    kws['flagtree_hints'] = ""
+                if flagtree_hints not in kws['flagtree_hints']:
+                    kws['flagtree_hints'] = flagtree_hints
 
     @staticmethod
     def maps_line_numbers_to_comment_hints(jit_fn):
@@ -49,4 +49,4 @@ def maps_line_numbers_to_comment_hints(jit_fn):
     @staticmethod
     def attach_line_number_to_comment_mapping(tree, line_flagtree_hints):
         if tree.body:
-            tree.body[0].line_flagtree_hints = line_flagtree_hints
+            tree.body[0].line_flagtree_hints = line_flagtree_hints