diff --git a/paddle/phi/backends/gpu/rocm/rocm_info.cc b/paddle/phi/backends/gpu/rocm/rocm_info.cc index b89d5a3c1624fca814f0fd2ef438944076b1de9c..1646d9666ff42c96db6386d93fa6d70d8506d4d0 100644 --- a/paddle/phi/backends/gpu/rocm/rocm_info.cc +++ b/paddle/phi/backends/gpu/rocm/rocm_info.cc @@ -301,14 +301,10 @@ bool IsGPUManagedMemorySupported(int dev_id) { "but received id is: %d. GPU count is: %d.", dev_id, GetGPUDeviceCount())); -#if defined(__linux__) || defined(_WIN32) - int ManagedMemoryAttr; - PADDLE_ENFORCE_GPU_SUCCESS(hipDeviceGetAttribute( - &ManagedMemoryAttr, hipDeviceAttributeManagedMemory, dev_id)); - return ManagedMemoryAttr != 0; -#else + // TODO(qili93): Hygon DTK (21.04 and 22.04) not support + // hipDeviceAttributeManagedMemory, temporary disable by default, to be + // verified in next DTK release return false; -#endif } bool IsGPUManagedMemoryOversubscriptionSupported(int dev_id) {