diff --git a/aten/src/ATen/miopen/Descriptors.h b/aten/src/ATen/miopen/Descriptors.h index a0ad4a4e1098a..2eee837cd533d 100644 --- a/aten/src/ATen/miopen/Descriptors.h +++ b/aten/src/ATen/miopen/Descriptors.h @@ -39,7 +39,7 @@ struct DescriptorDeleter { // function. template // NOLINTNEXTLINE(bugprone-exception-escape) -class TORCH_CUDA_CPP_API Descriptor { +class TORCH_HIP_CPP_API Descriptor { public: // Use desc() to access the underlying descriptor pointer in // a read-only fashion. Most client code should use this. @@ -65,7 +65,7 @@ class TORCH_CUDA_CPP_API Descriptor { std::unique_ptr> desc_; }; -class TORCH_CUDA_CPP_API TensorDescriptor : public Descriptor< +class TORCH_HIP_CPP_API TensorDescriptor : public Descriptor< miopenTensorDescriptor, &miopenCreateTensorDescriptor, &miopenDestroyTensorDescriptor> { @@ -88,7 +88,7 @@ class TORCH_CUDA_CPP_API TensorDescriptor : public Descriptor< std::ostream& operator<<(std::ostream & out, const TensorDescriptor& d); -class TORCH_CUDA_CPP_API FilterDescriptor : public Descriptor< +class TORCH_HIP_CPP_API FilterDescriptor : public Descriptor< miopenTensorDescriptor, &miopenCreateTensorDescriptor, &miopenDestroyTensorDescriptor> { @@ -105,7 +105,7 @@ class TORCH_CUDA_CPP_API FilterDescriptor : public Descriptor< } }; -struct TORCH_CUDA_CPP_API ConvolutionDescriptor +struct TORCH_HIP_CPP_API ConvolutionDescriptor : public Descriptor< miopenConvolutionDescriptor, &miopenCreateConvolutionDescriptor, @@ -121,7 +121,7 @@ struct TORCH_CUDA_CPP_API ConvolutionDescriptor }; // NOLINTNEXTLINE(bugprone-exception-escape) -struct TORCH_CUDA_CPP_API DropoutDescriptor +struct TORCH_HIP_CPP_API DropoutDescriptor : public Descriptor< miopenDropoutDescriptor, &miopenCreateDropoutDescriptor, @@ -137,7 +137,7 @@ struct TORCH_CUDA_CPP_API DropoutDescriptor } }; -struct TORCH_CUDA_CPP_API RNNDescriptor +struct TORCH_HIP_CPP_API RNNDescriptor : public Descriptor diff --git a/aten/src/ATen/miopen/Handle.h b/aten/src/ATen/miopen/Handle.h index 4c80c3aea65bf..b1637fca0a582 100644 --- a/aten/src/ATen/miopen/Handle.h +++ b/aten/src/ATen/miopen/Handle.h @@ -5,5 +5,5 @@ namespace at::native { -TORCH_CUDA_CPP_API miopenHandle_t getMiopenHandle(); +TORCH_HIP_CPP_API miopenHandle_t getMiopenHandle(); } // namespace at::native diff --git a/aten/src/ATen/miopen/Types.h b/aten/src/ATen/miopen/Types.h index 0a8a1a952e2e2..fdc0f6a607b71 100644 --- a/aten/src/ATen/miopen/Types.h +++ b/aten/src/ATen/miopen/Types.h @@ -6,7 +6,7 @@ namespace at::native { -TORCH_CUDA_CPP_API miopenDataType_t getMiopenDataType(const at::Tensor& tensor); +TORCH_HIP_CPP_API miopenDataType_t getMiopenDataType(const at::Tensor& tensor); int64_t miopen_version();