diff --git a/src/gpu/intel/ocl/ocl_math_utils.h b/src/gpu/intel/ocl/ocl_math_utils.h index 073ceb055f8..2433d5eda9a 100644 --- 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