xe: ocl: add missing type enablement

oneapi-src · Jan 22, 2025 · 87d098e · 87d098e
1 parent ac3ca50
commit 87d098e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/gpu/intel/ocl/ocl_math_utils.h b/src/gpu/intel/ocl/ocl_math_utils.h
@@ -45,13 +45,15 @@ int rnd_down(int a, unsigned int b) {
 #pragma OPENCL EXTENSION cl_khr_fp16 : enable
 
 #if DT_BF8 || SRC_DT_BF8 || WEI_DT_BF8 || DST_DT_BF8 || BIA_DT_BF8 || A_DT_BF8 \
-        || B_DT_BF8 || C_DT_BF8 || DATA_DT_BF8 || POST_OP_USING_BF8
+        || B_DT_BF8 || C_DT_BF8 || DATA_DT_BF8 || POST_OP_USING_BF8 \
+        || SRC_SCALES_DT_BF8 || WEI_SCALES_DT_BF8 || DST_SCALES_DT_BF8
 #define MATH_UTILS_DECLARE_BF8 1
 #endif
 
 #if DT_HF8 || SRC_DT_HF8 || WEI_DT_HF8 || DST_DT_HF8 || BIA_DT_HF8 || A_DT_HF8 \
         || A_DT_HF8 || B_DT_HF8 || C_DT_HF8 || DATA_DT_HF8 \
-        || POST_OP_USING_HF8
+        || POST_OP_USING_HF8 || SRC_SCALES_DT_HF8 || WEI_SCALES_DT_HF8 \
+        || DST_SCALES_DT_HF8
 #define MATH_UTILS_DECLARE_HF8 1
 #endif