xe: jit: gemm: revert handling for f32 src, f16 fpmath

oneapi-src · Jan 21, 2025 · 146ea11 · 146ea11
1 parent 22037c4
commit 146ea11
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/gpu/intel/jit/gemm/gen_gemm_kernel.cpp b/src/gpu/intel/jit/gemm/gen_gemm_kernel.cpp
@@ -574,9 +574,7 @@ status_t gen_gemm_nocopy_kernel_desc_t::select_kernel(compute::gpu_arch_t arch,
         return nullptr;
     });
 
-    if (fpmath_bf16
-            && (utils::one_of(Type::f32, problem_.Ta, problem_.Tb)
-                    || (problem_.Ta.isF8() || problem_.Tb.isF8()))
+    if (fpmath_bf16 && (problem_.Ta.isF8() || problem_.Tb.isF8())
             && (problem_.Ta.isInteger() || problem_.Tb.isInteger())) {
         if (problem_.Ta.isInt8() || problem_.Ta.isInt4()) {
             match_params.emplace_back(match_params[0]);
@@ -587,9 +585,7 @@ status_t gen_gemm_nocopy_kernel_desc_t::select_kernel(compute::gpu_arch_t arch,
         }
     }
 
-    if (fpmath_f16
-            && (utils::one_of(Type::f32, problem_.Ta, problem_.Tb)
-                    || (problem_.Ta.isF8() || problem_.Tb.isF8()))
+    if (fpmath_f16 && (problem_.Ta.isF8() || problem_.Tb.isF8())
             && (problem_.Ta.isInteger() || problem_.Tb.isInteger())) {
         if (problem_.Ta.isInt8() || problem_.Ta.isInt4()) {
             match_params.emplace_back(match_params[0]);