[FRONTEND] Fixed bugs in global symbols resolution of @triton.jit'd functions (#136)

python/triton/code_gen.py

@@ -659,9 +659,12 @@ class JITFunction:
 
     def __call__(self, *args, generator: CodeGenerator, **meta):
         try:
+            gscope = generator.gscope.copy()
             lscope = generator.lscope.copy()
             values = generator.module.get_values().copy()
+            generator.gscope = sys.modules[self.fn.__module__].__dict__
             ret = generator.visit_FunctionDef(self.parse().body[0], inline=True, arg_values=args)
+            generator.gscope = gscope
             generator.lscope = lscope
             generator.module.set_values(values)
             return ret