File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed
aten/src/ATen/cuda/detail Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -151,10 +151,10 @@ bool CUDAHooks::isPinnedPtr(const void* data) const {
151
151
return false ;
152
152
}
153
153
#endif
154
- #if !defined(USE_ROCM)
155
- return attr.type == cudaMemoryTypeHost;
156
- #else
154
+ #if defined(USE_ROCM) && ROCM_VERSION < 60000
157
155
return attr.memoryType == cudaMemoryTypeHost;
156
+ #else
157
+ return attr.type == cudaMemoryTypeHost;
158
158
#endif
159
159
}
160
160
Original file line number Diff line number Diff line change @@ -86,10 +86,10 @@ namespace caffe2 {
86
86
class TensorCoreEngine {};
87
87
#endif // USE_ROCM
88
88
89
- #if !defined(USE_ROCM)
90
- #define CAFFE2_CUDA_PTRATTR_MEMTYPE type
91
- #else
89
+ #if defined(USE_ROCM) && ROCM_VERSION < 60000
92
90
#define CAFFE2_CUDA_PTRATTR_MEMTYPE memoryType
91
+ #else
92
+ #define CAFFE2_CUDA_PTRATTR_MEMTYPE type
93
93
#endif
94
94
95
95
/* *
You can’t perform that action at this time.
0 commit comments