update device compare

src/partition/ndarray_partition.cc

@@ -107,7 +107,7 @@ class RangePartition : public NDArrayPartition {
         // we have only one CPU context, and can safely copy the array to that.
         range_cpu_(range.CopyTo(DGLContext{kDGLCPU, 0})) {
     auto ctx = range->ctx;
-    if (ctx.device_type != kDGLCUDA || ctx.device_type != kDGLROCM) {
+    if (ctx.device_type != kDGLCUDA && ctx.device_type != kDGLROCM) {
       LOG(FATAL) << "The range for an NDArrayPartition is only supported "
                     " on GPUs. Transfer the range to the target device before "
                     "creating the partition.";