update src and graphbolt code

src/array/libra_partition.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2021 Intel Corporation
  *
@@ -21,7 +22,7 @@
 #include <vector>
 
 #include "../c_api_common.h"
-#include "./check.h"
+#include "check.h"
 #include "kernel_decl.h"
 
 using namespace dgl::runtime;

src/array/selector.h

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2020 by Contributors
  * @file array/selector.h
@@ -12,13 +13,13 @@ namespace dgl {
 
 namespace {
 
-#ifdef __CUDACC__
-#define DGLDEVICE __device__
+#ifdef __HIPCC__
+#define DGLDEVICE __device__ __host__
 #define DGLINLINE __forceinline__
 #else
 #define DGLDEVICE
 #define DGLINLINE inline
-#endif  // __CUDACC__
+#endif  // __HIPCC__
 
 }  // namespace
 

src/array/uvm_array.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2019-2022 by Contributors
  * @file array/uvm_array.cc
@@ -8,7 +9,7 @@
 #include <sstream>
 
 #include "../c_api_common.h"
-#include "./uvm_array_op.h"
+#include "uvm_array_op.h"
 
 using namespace dgl::runtime;
 

src/geometry/cuda/edge_coarsening_impl.cu → src/geometry/cuda/edge_coarsening_impl.hip

+// !!! This is a file automatically generated by hipify!!!
+#include "hip/hip_runtime.h"
 /**
  *  Copyright (c) 2019 by Contributors
  * @file geometry/cuda/edge_coarsening_impl.cu
  * @brief Edge coarsening CUDA implementation
  */
-#include <curand_kernel.h>
+#include <hiprand/hiprand_kernel.h>
 #include <dgl/array.h>
 #include <dgl/random.h>
 #include <dmlc/thread_local.h>
@@ -32,9 +34,9 @@ __global__ void generate_uniform_kernel(
     float *ret_values, size_t num, uint64_t seed) {
   size_t id = blockIdx.x * blockDim.x + threadIdx.x;
   if (id < num) {
-    curandState state;
-    curand_init(seed, id, 0, &state);
-    ret_values[id] = curand_uniform(&state);
+    hiprandState_t state;
+    hiprand_init(seed, id, 0, &state);
+    ret_values[id] = hiprand_uniform(&state);
   }
 }
 
@@ -116,7 +118,7 @@ __global__ void weighted_respond_kernel(
 template <typename IdType>
 bool Colorize(IdType *result_data, int64_t num_nodes, float *const prop) {
   // initial done signal
-  cudaStream_t stream = runtime::getCurrentCUDAStream();
+  hipStream_t stream = runtime::getCurrentHIPStreamMasqueradingAsCUDA();
   CUDA_KERNEL_CALL(init_done_kernel, 1, 1, 0, stream);
 
   // generate color prop for each node
@@ -132,8 +134,8 @@ bool Colorize(IdType *result_data, int64_t num_nodes, float *const prop) {
       colorize_kernel, num_blocks, num_threads, 0, stream, prop, num_nodes,
       result_data);
   bool done_h = false;
-  CUDA_CALL(cudaMemcpyFromSymbol(
-      &done_h, done_d, sizeof(done_h), 0, cudaMemcpyDeviceToHost));
+  CUDA_CALL(hipMemcpyFromSymbol(
+      &done_h, done_d, sizeof(done_h), 0, hipMemcpyDeviceToHost));
   return done_h;
 }
 
@@ -155,7 +157,7 @@ bool Colorize(IdType *result_data, int64_t num_nodes, float *const prop) {
 template <DGLDeviceType XPU, typename FloatType, typename IdType>
 void WeightedNeighborMatching(
     const aten::CSRMatrix &csr, const NDArray weight, IdArray result) {
-  cudaStream_t stream = runtime::getCurrentCUDAStream();
+  hipStream_t stream = runtime::getCurrentHIPStreamMasqueradingAsCUDA();
   const auto &ctx = result->ctx;
   auto device = runtime::DeviceAPI::Get(ctx);
   device->SetDevice(ctx);
@@ -216,7 +218,7 @@ void NeighborMatching(const aten::CSRMatrix &csr, IdArray result) {
   device->SetDevice(ctx);
 
   // generate random weights
-  cudaStream_t stream = runtime::getCurrentCUDAStream();
+  hipStream_t stream = runtime::getCurrentHIPStreamMasqueradingAsCUDA();
   NDArray weight = NDArray::Empty(
       {num_edges}, DGLDataType{kDGLFloat, sizeof(float) * 8, 1}, ctx);
   float *weight_data = static_cast<float *>(weight->data);

src/geometry/cuda/geometry_op_impl.cu → src/geometry/cuda/geometry_op_impl.hip

+// !!! This is a file automatically generated by hipify!!!
+#include "hip/hip_runtime.h"
 /**
  *  Copyright (c) 2019 by Contributors
  * @file geometry/cuda/geometry_op_impl.cc
@@ -95,7 +97,7 @@ template <DGLDeviceType XPU, typename FloatType, typename IdType>
 void FarthestPointSampler(
     NDArray array, int64_t batch_size, int64_t sample_points, NDArray dist,
     IdArray start_idx, IdArray result) {
-  cudaStream_t stream = runtime::getCurrentCUDAStream();
+  hipStream_t stream = runtime::getCurrentHIPStreamMasqueradingAsCUDA();
 
   const FloatType* array_data = static_cast<FloatType*>(array->data);
 
@@ -110,7 +112,7 @@ void FarthestPointSampler(
 
   // sample for each cloud in the batch
   IdType* start_idx_data = static_cast<IdType*>(start_idx->data);
-  CUDA_CALL(cudaSetDevice(array->ctx.device_id));
+  CUDA_CALL(hipSetDevice(array->ctx.device_id));
 
   CUDA_KERNEL_CALL(
       fps_kernel, batch_size, THREADS, 0, stream, array_data, batch_size,

src/geometry/geometry.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2019 by Contributors
  * @file geometry/geometry.cc
@@ -10,7 +11,7 @@
 
 #include "../array/check.h"
 #include "../c_api_common.h"
-#include "./geometry_op.h"
+#include "geometry_op.h"
 
 using namespace dgl::runtime;
 

src/graph/creators.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2020 by Contributors
  * @file graph/creators.cc
  * @brief Functions for constructing graphs.
  */
-#include "./heterograph.h"
+#include "heterograph.h"
 using namespace dgl::runtime;
 
 namespace dgl {

src/graph/heterograph.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2019 by Contributors
  * @file graph/heterograph.cc
  * @brief Heterograph implementation
  */
-#include "./heterograph.h"
+#include "heterograph.h"
 
 #include <dgl/array.h>
 #include <dgl/graph_serializer.h>

src/graph/heterograph.h

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2019 by Contributors
  * @file graph/heterograph.h
@@ -18,7 +19,7 @@
 #include <utility>
 #include <vector>
 
-#include "./unit_graph.h"
+#include "unit_graph.h"
 #include "shared_mem_manager.h"
 
 namespace dgl {

src/graph/heterograph_capi.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2020 by Contributors
  * @file graph/heterograph_capi.cc
@@ -14,7 +15,7 @@
 #include <set>
 
 #include "../c_api_common.h"
-#include "./heterograph.h"
+#include "heterograph.h"
 #include "unit_graph.h"
 
 using namespace dgl::runtime;

src/graph/pickle.cc

+// !!! This is a file automatically generated by hipify!!!
 /**
  *  Copyright (c) 2020 by Contributors
  * @file graph/pickle.cc
@@ -10,7 +11,7 @@
 #include <dmlc/memory_io.h>
 
 #include "../c_api_common.h"
-#include "./heterograph.h"
+#include "heterograph.h"
 #include "unit_graph.h"
 
 using namespace dgl::runtime;