File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -1537,11 +1537,8 @@ void scaled_gemm(
15371537 // rowwise isn't supported using cublaslt or older hipblaslt
15381538 TORCH_INTERNAL_ASSERT (use_rowwise == false , " rowwise scaled_gemm not supported with blaslt" );
15391539#endif
1540- // do not remove {}, this scope corresponds to the else condition in the USE_ROCM section above
1541- {
1542- computeDesc.setAttribute (CUBLASLT_MATMUL_DESC_A_SCALE_POINTER, mat1_scale_ptr);
1543- computeDesc.setAttribute (CUBLASLT_MATMUL_DESC_B_SCALE_POINTER, mat2_scale_ptr);
1544- }
1540+ computeDesc.setAttribute (CUBLASLT_MATMUL_DESC_A_SCALE_POINTER, mat1_scale_ptr);
1541+ computeDesc.setAttribute (CUBLASLT_MATMUL_DESC_B_SCALE_POINTER, mat2_scale_ptr);
15451542 if (result_scale_ptr != nullptr ) {
15461543 computeDesc.setAttribute (CUBLASLT_MATMUL_DESC_D_SCALE_POINTER, result_scale_ptr);
15471544 }
You can’t perform that action at this time.
0 commit comments