Add TRITON_INTEL_DEVICE_ARCH env var to redefine architecture if needed (#4673)

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

python/triton/knobs.py

@@ -506,6 +506,7 @@ class intel_knobs(base_knobs):
     # `ocloc query CL_DEVICE_EXTENSIONS`. If not set, a compiler calls `ocloc` in runtime to get
     # the actual device extensions.
     device_extensions: env_opt_str = env_opt_str("TRITON_INTEL_DEVICE_EXTENSIONS")
+    device_arch: env_opt_str = env_opt_str("TRITON_INTEL_DEVICE_ARCH")
 
 
 class amd_knobs(base_knobs):

third_party/intel/backend/compiler.py

@@ -118,7 +118,7 @@ def __init__(self, target: tuple) -> None:
             raise TypeError("target.arch is not a dict")
         dirname = os.path.dirname(os.path.realpath(__file__))
         mod = compile_module_from_src(Path(os.path.join(dirname, "arch_parser.c")).read_text(), "arch_utils")
-        self.device_arch = mod.parse_device_arch(target.arch.get('architecture', 0))
+        self.device_arch = knobs.intel.device_arch or mod.parse_device_arch(target.arch.get('architecture', 0))
         self.properties = self.parse_target(target.arch)
         self.binary_ext = "spv"