修复c++测试用例中有关于kDLROCM的问题

tests/cpp/common.h

@@ -6,7 +6,7 @@
 static constexpr DLContext CTX = DLContext{kDLCPU, 0};
 static constexpr DLContext CPU = DLContext{kDLCPU, 0};
 #ifdef DGL_USE_CUDA
-static constexpr DLContext GPU = DLContext{kDLGPU, 0};
+static constexpr DLContext GPU = DLContext{kDLROCM, 0};
 #endif
 
 template <typename T>

tests/cpp/test_partition.cc

@@ -80,11 +80,11 @@ void _TestRemainder_MapToX() {
 
 TEST(PartitionTest, TestRemainderPartition) {
 #ifdef DGL_USE_CUDA
-  _TestRemainder_GeneratePermutation<kDLGPU, int32_t>();
-  _TestRemainder_GeneratePermutation<kDLGPU, int64_t>();
+  _TestRemainder_GeneratePermutation<kDLROCM, int32_t>();
+  _TestRemainder_GeneratePermutation<kDLROCM, int64_t>();
 
-  _TestRemainder_MapToX<kDLGPU, int32_t>();
-  _TestRemainder_MapToX<kDLGPU, int64_t>();
+  _TestRemainder_MapToX<kDLROCM, int32_t>();
+  _TestRemainder_MapToX<kDLROCM, int64_t>();
 #endif
   // CPU is not implemented
 }
@@ -189,11 +189,11 @@ void _TestRange_MapToX() {
 
 TEST(PartitionTest, TestRangePartition) {
 #ifdef DGL_USE_CUDA
-  _TestRange_GeneratePermutation<kDLGPU, int32_t>();
-  _TestRange_GeneratePermutation<kDLGPU, int64_t>();
+  _TestRange_GeneratePermutation<kDLROCM, int32_t>();
+  _TestRange_GeneratePermutation<kDLROCM, int64_t>();
 
-  _TestRange_MapToX<kDLGPU, int32_t>();
-  _TestRange_MapToX<kDLGPU, int64_t>();
+  _TestRange_MapToX<kDLROCM, int32_t>();
+  _TestRange_MapToX<kDLROCM, int64_t>();
 #endif
   // CPU is not implemented
 }