diff --git a/csrc/quantization/fp8/amd/quant_utils.cuh b/csrc/quantization/fp8/amd/quant_utils.cuh index b2196b8e..b812b28b 100644 --- a/csrc/quantization/fp8/amd/quant_utils.cuh +++ b/csrc/quantization/fp8/amd/quant_utils.cuh @@ -24,12 +24,12 @@ __inline__ __device__ Tout scaled_vec_conversion(const Tin& x, return x; } - #if HIP_FP8_TYPE_FNUZ -using fp8_type = __hip_fp8_e4m3_fnuz; -using fp8x2_type = __hip_fp8x2_e4m3_fnuz; - #elif HIP_FP8_TYPE_OCP + #if HIP_FP8_TYPE_OCP using fp8_type = __hip_fp8_e4m3; using fp8x2_type = __hip_fp8x2_e4m3; + #else +using fp8_type = __hip_fp8_e4m3_fnuz; +using fp8x2_type = __hip_fp8x2_e4m3_fnuz; #endif // fp8 -> half