From 6230b5f6b3298cf0f97ec1413c6968078c4d0bd4 Mon Sep 17 00:00:00 2001
From: Bertan Dogancay <111835151+BertanDogancay@users.noreply.github.com>
Date: Tue, 5 Sep 2023 09:44:12 -0600
Subject: [PATCH] RCCL should use hipPointerAttribute_t.type (#872)

---
 src/misc/argcheck.cc | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/src/misc/argcheck.cc b/src/misc/argcheck.cc
index 994d1fd9b1..84f8a6f0d5 100644
--- a/src/misc/argcheck.cc
+++ b/src/misc/argcheck.cc
@@ -14,11 +14,7 @@ static ncclResult_t CudaPtrCheck(const void* pointer, struct ncclComm* comm, con
     WARN("%s : %s %p is not a valid pointer", opname, ptrname, pointer);
     return ncclInvalidArgument;
   }
-#if CUDART_VERSION >= 10000
   if (attr.type == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {
-#else
-  if (attr.memoryType == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {
-#endif
     WARN("%s : %s allocated on device %d mismatchs with NCCL device %d", opname, ptrname, attr.device, comm->cudaDev);
     return ncclInvalidArgument;
   }