Change SPV_KHR_bfloat16 workaround patch to work for different drivers

Signed-off-by: Whitney Tsang <[email protected]>

Author: whitneywhtsang

setup.py

@@ -489,15 +489,6 @@ def build_extension(self, ext):
             cmake_args.append("-DLLVM_EXTERNAL_LIT=" + lit_dir)
         cmake_args.extend(thirdparty_cmake_args)
 
-        result = subprocess.run(["bash", "./scripts/capture-hw-details.sh"], stdout=subprocess.PIPE,
-                                stderr=subprocess.PIPE, check=True, text=True, env=os.environ.copy())
-        agama_version = None
-        for line in result.stdout.splitlines():
-            if line.startswith("AGAMA_VERSION="):
-                agama_version = line.split("=", 1)[1].strip()
-                break
-        cmake_args.append(f"-DAGAMA_VERSION={agama_version}")
-
         # configuration
         cfg = get_build_type()
         build_args = ["--config", cfg]

third_party/intel/cmake/3122.patch

@@ -1,20 +1,27 @@
 diff --git a/lib/SPIRV/SPIRVWriter.cpp b/lib/SPIRV/SPIRVWriter.cpp
-index a124ba48c..3f46b5685 100644
+index ec4ec41f..9cad08d8 100644
 --- a/lib/SPIRV/SPIRVWriter.cpp
 +++ b/lib/SPIRV/SPIRVWriter.cpp
-@@ -397,6 +397,7 @@ SPIRVType *LLVMToSPIRVBase::transType(Type *T) {
-     }
+@@ -401,13 +401,15 @@ SPIRVType *LLVMToSPIRVBase::transType(Type *T) {
    }
 
-+#if 0
    if (T->isBFloatTy()) {
-     BM->getErrorLog().checkError(
-         BM->isAllowedToUseExtension(ExtensionID::SPV_KHR_bfloat16),
-@@ -406,6 +407,7 @@ SPIRVType *LLVMToSPIRVBase::transType(Type *T) {
-         "requires this extension");
-     return mapType(T, BM->addFloatType(16, FPEncodingBFloat16KHR));
+-    BM->getErrorLog().checkError(
+-        BM->isAllowedToUseExtension(ExtensionID::SPV_KHR_bfloat16),
+-        SPIRVEC_RequiresExtension,
+-        "SPV_KHR_bfloat16\n"
+-        "NOTE: LLVM module contains bfloat type, translation of which "
+-        "requires this extension");
+-    return mapType(T, BM->addFloatType(16, FPEncodingBFloat16KHR));
++    if (BM->isAllowedToUseExtension(ExtensionID::SPV_KHR_bfloat16)) {
++      BM->getErrorLog().checkError(
++          BM->isAllowedToUseExtension(ExtensionID::SPV_KHR_bfloat16),
++          SPIRVEC_RequiresExtension,
++          "SPV_KHR_bfloat16\n"
++          "NOTE: LLVM module contains bfloat type, translation of which "
++          "requires this extension");
++      return mapType(T, BM->addFloatType(16, FPEncodingBFloat16KHR));
++    }
    }
-+#endif
 
    if (T->isFloatingPointTy())
-     return mapType(T, BM->addFloatType(T->getPrimitiveSizeInBits()));

third_party/intel/cmake/FindSPIRVToLLVMTranslator.cmake

@@ -27,29 +27,27 @@ if (NOT SPIRVToLLVMTranslator_FOUND)
             FetchContent_MakeAvailable(spirv-llvm-translator)
 
             # FIXME: Don't apply patch when LTS driver is updated.
-            if(DEFINED AGAMA_VERSION AND AGAMA_VERSION STREQUAL "1146")
+            execute_process(
+                COMMAND git apply --check ${CMAKE_CURRENT_LIST_DIR}/3122.patch
+                WORKING_DIRECTORY ${spirv-llvm-translator_SOURCE_DIR}
+                ERROR_QUIET
+                RESULT_VARIABLE PATCH_RESULT
+            )
+            if(PATCH_RESULT EQUAL 0)
                 execute_process(
-                    COMMAND git apply --check ${CMAKE_CURRENT_LIST_DIR}/3122.patch
-                    WORKING_DIRECTORY ${spirv-llvm-translator_SOURCE_DIR}
-                    ERROR_QUIET
-                    RESULT_VARIABLE PATCH_RESULT
+                        COMMAND git apply ${CMAKE_CURRENT_LIST_DIR}/3122.patch
+                        WORKING_DIRECTORY ${spirv-llvm-translator_SOURCE_DIR}
+                        RESULT_VARIABLE PATCH_RESULT
                 )
-                if(PATCH_RESULT EQUAL 0)
-                    execute_process(
-                            COMMAND git apply ${CMAKE_CURRENT_LIST_DIR}/3122.patch
-                            WORKING_DIRECTORY ${spirv-llvm-translator_SOURCE_DIR}
-                            RESULT_VARIABLE PATCH_RESULT
-                    )
-                else()
-                    execute_process( # Check if the patch is already applied
-                            COMMAND git apply --reverse --check ${CMAKE_CURRENT_LIST_DIR}/3122.patch
-                            WORKING_DIRECTORY ${spirv-llvm-translator_SOURCE_DIR}
-                            RESULT_VARIABLE PATCH_RESULT
-                    )
-                endif()
-                if(NOT PATCH_RESULT EQUAL 0)
-                    message(FATAL_ERROR "Failed to apply 3122.patch to SPIRV-LLVM-Translator")
-                endif()
+            else()
+                execute_process( # Check if the patch is already applied
+                        COMMAND git apply --reverse --check ${CMAKE_CURRENT_LIST_DIR}/3122.patch
+                        WORKING_DIRECTORY ${spirv-llvm-translator_SOURCE_DIR}
+                        RESULT_VARIABLE PATCH_RESULT
+                )
+            endif()
+            if(NOT PATCH_RESULT EQUAL 0)
+                message(FATAL_ERROR "Failed to apply 3122.patch to SPIRV-LLVM-Translator")
             endif()
 
             # FIXME: Don't apply patch when Agama driver is updated to incorporate with the SPV_INTEL_bfloat16_arithmetic extension.