[TRITONGPU] Added template for Triton -> TritonGPU conversion
This commit is contained in:
@@ -1307,6 +1307,15 @@ class JITFunction:
|
||||
raise CompilationError(self.src, node) from e
|
||||
# FIXME: now we need to return context, otherwise it will be deleted
|
||||
return generator.module, context
|
||||
|
||||
def compile_ttir_to_llir(self, mod, ctx):
|
||||
pm = _triton.ir.pass_manager(ctx)
|
||||
pm.add_inliner_pass()
|
||||
pm.add_triton_combine_pass()
|
||||
pm.add_canonicalizer_pass()
|
||||
pm.add_convert_triton_to_tritongpu_pass()
|
||||
pm.run(mod)
|
||||
return mod
|
||||
|
||||
|
||||
def __getitem__(self, grid):
|
||||
|
Reference in New Issue
Block a user