[intel] Align 'compile_module_from_src' with other backends after 'cfc5b76' commit

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

Author: anmyachev

third_party/intel/backend/driver.py

@@ -11,7 +11,7 @@
 from triton.runtime.build import _build
 from triton.runtime.cache import get_cache_manager
 from triton.backends.compiler import GPUTarget
-from triton.backends.driver import DriverBase
+from triton.backends.driver import DriverBase, platform_key
 
 # A hard-coded cache version that can be updated when we know that the cached file is invalid and
 # there are no other ways to detect that the runtime environment has changed. For example, a shared
@@ -251,11 +251,11 @@ def __del__(self):
 
 def compile_module_from_src(src, name):
     hasher = hashlib.sha256(__CACHE_VERSION.encode("utf-8"))
-    hasher.update(src.encode("utf-8"))
+    hasher.update((src + platform_key()).encode("utf-8"))
     key = hasher.hexdigest()
     cache = get_cache_manager(key)
-    file_name = f"{name}.{sysconfig.get_config_var('EXT_SUFFIX').split('.')[-1]}"
-    cache_path = cache.get_file(file_name)
+    suffix = sysconfig.get_config_var("EXT_SUFFIX")
+    cache_path = cache.get_file(f"{name}{suffix}")
     if cache_path is None:
         with tempfile.TemporaryDirectory() as tmpdir:
             src_path = os.path.join(tmpdir, "main.cpp")
@@ -271,7 +271,7 @@ def compile_module_from_src(src, name):
             so = _build(name, src_path, tmpdir, COMPILATION_HELPER.library_dir, COMPILATION_HELPER.include_dir,
                         COMPILATION_HELPER.libraries, extra_compile_args=extra_compiler_args)
             with open(so, "rb") as f:
-                cache_path = cache.put(f.read(), file_name, binary=True)
+                cache_path = cache.put(f.read(), f"{name}{suffix}", binary=True)
 
     if name == 'arch_utils':
         return ArchParser(cache_path)