diff --git a/torchao/csrc/cuda/rowwise_scaled_linear_cutlass/rowwise_scaled_linear_cutlass.cuh b/torchao/csrc/cuda/rowwise_scaled_linear_cutlass/rowwise_scaled_linear_cutlass.cuh index 058cec4ba4..f920c36714 100644 --- a/torchao/csrc/cuda/rowwise_scaled_linear_cutlass/rowwise_scaled_linear_cutlass.cuh +++ b/torchao/csrc/cuda/rowwise_scaled_linear_cutlass/rowwise_scaled_linear_cutlass.cuh @@ -214,7 +214,7 @@ void rowwise_scaled_linear_kernel_cutlass_sm8x( ElementBias(0), {cute::_0{}, cute::_1{}, problem_size.n()}}; } else { - return {ElementBias(0)}; + return {{ElementBias(0)}}; } }() }; diff --git a/torchao/csrc/cuda/rowwise_scaled_linear_sparse_cutlass/rowwise_scaled_linear_sparse_cutlass.cuh b/torchao/csrc/cuda/rowwise_scaled_linear_sparse_cutlass/rowwise_scaled_linear_sparse_cutlass.cuh index e61a2d58b6..4342919d6e 100644 --- a/torchao/csrc/cuda/rowwise_scaled_linear_sparse_cutlass/rowwise_scaled_linear_sparse_cutlass.cuh +++ b/torchao/csrc/cuda/rowwise_scaled_linear_sparse_cutlass/rowwise_scaled_linear_sparse_cutlass.cuh @@ -193,7 +193,7 @@ void rowwise_scaled_linear_sparse_kernel_cutlass_sm9x( {cute::_1{}, cute::_0{}, cute::_0{}} }; } else { - return {ElementBias(0)}; + return {{ElementBias(0)}}; } }() };