Skip to content

Commit

Permalink
Revert "Fix issue with setAttribute and int8_t vs int32_t variables (p…
Browse files Browse the repository at this point in the history
…ytorch#143693)"

This reverts commit ae3d385.

Reverted pytorch#143693 on behalf of https://github.com/huydhn due to Sorry for reverting this change but it has a conflict with pytorch#143639 that is breaking trunk ([comment](pytorch#143693 (comment)))
  • Loading branch information
pytorchmergebot committed Dec 21, 2024
1 parent a8953c3 commit b89bfe0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aten/src/ATen/cuda/CUDABlas.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1453,7 +1453,7 @@ void scaled_gemm(
computeDesc.setAttribute(CUBLASLT_MATMUL_DESC_D_SCALE_POINTER, result_scale_ptr);
}
#ifndef USE_ROCM
computeDesc.setAttribute(CUBLASLT_MATMUL_DESC_FAST_ACCUM, use_fast_accum);
computeDesc.setAttribute(CUBLASLT_MATMUL_DESC_FAST_ACCUM, use_fast_accum ? 1 : 0);
#endif
CuBlasLtMatrixLayout Adesc(ScalarTypeToCudaDataType(mat1_dtype), m, k, mat1_ld, transa == 't');
CuBlasLtMatrixLayout Bdesc(ScalarTypeToCudaDataType(mat2_dtype), k, n, mat2_ld, transb == 't');
Expand Down

0 comments on commit b89bfe0

Please sign in to comment.