bench_UVM_GIN.py

#!/usr/bin/env python3
import os
import sys

os.environ["PYTHONWARNINGS"] = "ignore"
os.environ["LD_LIBRARY_PATH"] += os.pathsep + 'local/openmpi-4.1.1/lib/'
os.environ["PATH"] += os.pathsep + 'local/openmpi-4.1.1/openmpi-4.1.1/bin/'
os.environ["LD_LIBRARY_PATH"] += os.pathsep + 'local/cudnn-v8.2/lib64'

hidden = 64
num_GPUs = int(sys.argv[1])

dataset = [
            ( 'Reddit'                      , 602      	, 41),
            ( 'enwiki-2013'	            , 300	, 12),   
            ( 'it-2004'                     , 256       , 64),
            ( 'paper100M'                   , 128       , 64),
            ( 'ogbn-products'	            , 100	, 47),   
            ( 'ogbn-proteins'	            , 8		, 112),
            ( 'com-Orkut'		    , 128       , 32),
]

GPU_avail = "CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 "
# command = GPU_avail + "build/uvm_gin_5layer "
command = GPU_avail + "build/uvm_gin_5layer_mem "

for data, inDim, classes in dataset:
        beg_file = "dataset/bin/{}_beg_pos.bin".format(data)
        csr_file = "dataset/bin/{}_csr.bin".format(data)
        weight_file = "dataset/bin/{}_weight.bin".format(data)
        os.system("{0} {1} {2} {3} {4} {5} {6} {7}".\
                format(command, beg_file, csr_file, weight_file, 
                        num_GPUs, inDim, hidden, classes))