diff options
| author | bddppq <bai@in.tum.de> | 2019-04-05 11:09:15 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-05 11:15:16 -0700 |
| commit | c3a559deb72626b664729192fd551f9bf269a43c (patch) | |
| tree | 656ee2e2c03b58ac4c7d6cdd43a862d53ab9489f | |
| parent | fefa6d305ea3e820afe64cec015d2f6746d9ca88 (diff) | |
| download | pytorch-c3a559deb72626b664729192fd551f9bf269a43c.tar.gz pytorch-c3a559deb72626b664729192fd551f9bf269a43c.tar.bz2 pytorch-c3a559deb72626b664729192fd551f9bf269a43c.zip | |
Remove cuda::compat functions in aten (#18905)
Summary:
Looks like the issue of using `std::` functions is fixed in new rocm version
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18905
Differential Revision: D14792943
Pulled By: bddppq
fbshipit-source-id: af11acbb85872943f23b6e55415db1f0699e7b8f
| -rw-r--r-- | aten/src/ATen/native/Distributions.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/aten/src/ATen/native/Distributions.h b/aten/src/ATen/native/Distributions.h index 31167d5907..72b3ec9f67 100644 --- a/aten/src/ATen/native/Distributions.h +++ b/aten/src/ATen/native/Distributions.h @@ -21,7 +21,7 @@ static inline THGenerator* get_generator(at::Generator* gen) { }} // namespace at::native // ROCM hcc doesn't work well with using std:: in kernel functions -#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_HCC__) +#if defined(__CUDA_ARCH__) || defined(__HIPCC__) #include <c10/cuda/CUDAMathCompat.h> #define compat_exp c10::cuda::compat::exp #define compat_floor c10::cuda::compat::floor @@ -40,7 +40,7 @@ static inline THGenerator* get_generator(at::Generator* gen) { namespace { -#if !defined(__CUDA_ARCH__) && !defined(__HIP_PLATFORM_HCC__) +#if !defined(__CUDA_ARCH__) && !defined(__HIPCC__) // we cannot use std::isnan directly due to some incompatibility of // gcc constexpr'ing and nvcc #define isnan std::isnan |