Skip to content

Commit 42ad7e3

Browse files
[mx] fix build warning for mxfp8 dim1 cast CUDA kernel (#2782)
[mx] fix return type for 'quantize_block' in mxfp81 dim1 cast CUDA kernel
1 parent e6b38bb commit 42ad7e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/csrc/cuda/mx_kernels/mxfp8_quantize.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ __device__ __forceinline__ OType torchao_quantize_value(float input_value,
451451
* Template parameters ensure compile-time array size checking for safety
452452
*/
453453
template <typename OType, int NUM_VALUES, ScaleCalculationMode ScalingMode>
454-
__device__ __forceinline__ float
454+
__device__ __forceinline__ void
455455
quantize_block(float amax, e8m0_t &out_scale,
456456
const float (&input_values)[NUM_VALUES],
457457
OType (&output_values)[NUM_VALUES]) {

0 commit comments

Comments
 (0)