fix curand (#3077)

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

src/random/random.cc

@@ -24,6 +24,7 @@ DGL_REGISTER_GLOBAL("rng._CAPI_SetSeed")
 #pragma omp parallel for
     for (int i = 0; i < omp_get_max_threads(); ++i) {
       RandomEngine::ThreadLocal()->SetSeed(seed);
+    }
 #ifdef DGL_USE_CUDA
     auto* thr_entry = CUDAThreadEntry::ThreadLocal();
     if (!thr_entry->curand_gen) {
@@ -31,9 +32,8 @@ DGL_REGISTER_GLOBAL("rng._CAPI_SetSeed")
     }
     CURAND_CALL(curandSetPseudoRandomGeneratorSeed(
         thr_entry->curand_gen,
-          static_cast<uint64_t>(seed + GetThreadId())));
+        static_cast<uint64_t>(seed)));
 #endif  // DGL_USE_CUDA
-    }
   });
 
 DGL_REGISTER_GLOBAL("rng._CAPI_Choice")