| commit | ed0c0c49efeebbb8aa423945cf0840787cdf7c41 | [log] [tgz] |
|---|---|---|
| author | PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> | Tue Dec 19 01:01:42 2023 +0000 |
| committer | PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> | Tue Dec 19 01:01:42 2023 +0000 |
| tree | 194214546bb54aaf84e424ab7f821bbb54584886 | |
| parent | 368a0c06d4ed9fccb1e03cfa570c4d8530be9203 [diff] |
Revert "[ROCm] fix nightly 5.6 build (#116029)" This reverts commit 63e242b1e41759f9b24a0fbb997f157a06a9dd13. Reverted https://github.com/pytorch/pytorch/pull/116029 on behalf of https://github.com/jeanschmidt due to Need to revert, in order to be able to revert #114329 ([comment](https://github.com/pytorch/pytorch/pull/116029#issuecomment-1861931736))
diff --git a/aten/src/ATen/cuda/CUDABlas.cpp b/aten/src/ATen/cuda/CUDABlas.cpp index cc387dd..24c96f4 100644 --- a/aten/src/ATen/cuda/CUDABlas.cpp +++ b/aten/src/ATen/cuda/CUDABlas.cpp
@@ -1102,7 +1102,7 @@ #endif // (!defined(USE_ROCM) && !defined(_MSC_VER)) || (defined(USE_ROCM) && ROCM_VERSION >= 50700) // ROCm 5.6 hipblas matches the const Dtype *A API, but prior hipblas does not. -#if defined(USE_ROCM) && ROCM_VERSION < 50600 +#if defined(USE_ROCM) && ROCM_VERSION <= 50600 #define ROCM_CONST_BUG #else #define ROCM_CONST_BUG const
diff --git a/aten/src/ATen/cuda/CublasHandlePool.cpp b/aten/src/ATen/cuda/CublasHandlePool.cpp index 55a7be5..05c2ba9 100644 --- a/aten/src/ATen/cuda/CublasHandlePool.cpp +++ b/aten/src/ATen/cuda/CublasHandlePool.cpp
@@ -28,7 +28,7 @@ namespace { -#if defined(USE_ROCM) && ROCM_VERSION >= 50700 +#ifdef USE_ROCM void createCublasLtHandle(cublasLtHandle_t *handle) { TORCH_CUDABLAS_CHECK(cublasLtCreate(handle)); }