samplers little changes

install.sh

@@ -3,17 +3,16 @@
 mkdir -p build && cd build
 mkdir -p build && cd build
 cmake .. \
 cmake .. \
     -DCMAKE_EXPORT_COMPILE_COMMANDS=ON \
     -DCMAKE_EXPORT_COMPILE_COMMANDS=ON \
-<<<<<<< HEAD
-    -DCMAKE_PREFIX_PATH="/home/zlj/.miniconda3/envs/dgnn/lib/python3.10/site-packages" \
-    -DPython3_ROOT_DIR="/home/zlj/.miniconda3/envs/dgnn" \
-    -DCUDA_TOOLKIT_ROOT_DIR="/home/zlj/local/cuda-12.2" \
-=======
     -DCMAKE_PREFIX_PATH=$(python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())") \
     -DCMAKE_PREFIX_PATH=$(python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())") \
     -DPython3_ROOT_DIR=$(python -c "import sys; print(sys.prefix)") \
     -DPython3_ROOT_DIR=$(python -c "import sys; print(sys.prefix)") \
     -DCUDA_TOOLKIT_ROOT_DIR=${CUDA_HOME:-"$(realpath $(dirname $(which nvcc))/../)"} \
     -DCUDA_TOOLKIT_ROOT_DIR=${CUDA_HOME:-"$(realpath $(dirname $(which nvcc))/../)"} \
->>>>>>> 98ad16d40e5e3e0a7dcdbc4f21dc9e164abc625f
 && make -j32 \
 && make -j32 \
 && rm -rf ../starrygl/lib \
 && rm -rf ../starrygl/lib \
 && mkdir ../starrygl/lib \
 && mkdir ../starrygl/lib \
 && cp lib*.so ../starrygl/lib/ \
 && cp lib*.so ../starrygl/lib/ \
-&& patchelf --set-rpath '$ORIGIN:$ORIGIN/lib' --force-rpath ../starrygl/lib/*.so
+&& patchelf --set-rpath '$ORIGIN:$ORIGIN/lib' --force-rpath ../starrygl/lib/*.so
\ No newline at end of file
+
+
+    # -DCMAKE_PREFIX_PATH="/home/zlj/.miniconda3/envs/dgnn/lib/python3.10/site-packages" \
+    # -DPython3_ROOT_DIR="/home/zlj/.miniconda3/envs/dgnn" \
+    # -DCUDA_TOOLKIT_ROOT_DIR="/home/zlj/local/cuda-12.2" \
\ No newline at end of file

starrygl/sample/sample_core/neighbor_sampler.py

@@ -8,7 +8,7 @@ import torch.multiprocessing as mp
 from typing import Optional, Tuple
 from typing import Optional, Tuple
 
 
 from .base import BaseSampler, NegativeSampling, SampleOutput, SampleType
 from .base import BaseSampler, NegativeSampling, SampleOutput, SampleType
-# from sample_cores import ParallelSampler, get_neighbors, heads_unique
+from starrygl.lib.libstarrygl_sampler import ParallelSampler, get_neighbors, heads_unique
 
 
 from torch.distributed.rpc import rpc_async
 from torch.distributed.rpc import rpc_async
 
 

starrygl/sample/sample_core/reddit_demo_before.py

@@ -62,7 +62,7 @@ def test():
     torch.backends.cudnn.benchmark = False
     torch.backends.cudnn.benchmark = False
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.deterministic = True
 
 
-    g_data, df = load_gdelt_dataset()
+    g_data, df = load_reddit_dataset()
     print(g_data)
     print(g_data)
     # for worker in [1,2,3,4,5,6,7,8,9,10,20,30]:
     # for worker in [1,2,3,4,5,6,7,8,9,10,20,30]:
     # import random
     # import random

starrygl/sample/sample_core/reddit_demo_static.py

@@ -70,7 +70,7 @@ def test():
     torch.backends.cudnn.benchmark = False
     torch.backends.cudnn.benchmark = False
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.deterministic = True
 
 
-    g_data = load_ogb_dataset()
+    g_data = load_reddit_dataset()
     print(g_data)
     print(g_data)
 
 
     from .neighbor_sampler import NeighborSampler, get_neighbors
     from .neighbor_sampler import NeighborSampler, get_neighbors