use 'backend' instead of 'target'

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>

python/test/unit/tools/test_irsource.py

@@ -1,9 +1,10 @@
 import tempfile
 import triton
-from triton.compiler import IRSource
+from triton.compiler import IRSource, make_backend
 from triton._C.libtriton import ir
 
 target = triton.runtime.driver.active.get_current_target()
+backend = make_backend(target)
 
 
 def test_mlir_attribute_parsing() -> None:
@@ -41,7 +42,7 @@ def test_mlir_attribute_parsing() -> None:
         f.write(sample_ttgir)
         f.flush()
         context = ir.context()
-        src = IRSource(f.name, context, target)
+        src = IRSource(f.name, context, backend)
 
         # check name and type signature
         # should match ty_to_cpp(...)
@@ -87,7 +88,7 @@ def test_mlir_attribute_parsing() -> None:
         f.write(sample_ttgir_vector_add)
         f.flush()
         context = ir.context()
-        src = IRSource(f.name, context, target)
+        src = IRSource(f.name, context, backend)
 
         # now test compilation
         triton.compile(f.name, target=target)

python/triton/compiler/compiler.py

@@ -91,16 +91,12 @@ def parse_options(self):
 
 class IRSource:
 
-    def __init__(self, path, context, target=None):
+    def __init__(self, path, context, backend):
         self.path = path
         path = Path(path)
         self.ext = path.suffix[1:]
         self.src = path.read_text()
         ir.load_dialects(context)
-        if target is None:
-            target = driver.active.get_current_target()
-        assert isinstance(target, GPUTarget), "target must be of GPUTarget type"
-        backend = make_backend(target)
         backend.load_dialects(context)
 
         # We don't have a easy-to-use PTX parser that we can use, so keep that regex for now.
@@ -228,7 +224,7 @@ def compile(src, target=None, options=None):
     if ir_source:
         assert isinstance(src, str), "source must be either AST or a filepath"
         context = ir.context()
-        src = IRSource(src, context, target)
+        src = IRSource(src, context, backend)
 
     extra_options = src.parse_options()
     options = backend.parse_options(dict(options or dict(), **extra_options))