Skip to content

Latest commit

 

History

History

benchmark

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 

Run benchmark_kernel

This is for running cpp kernel under different configs.

First please do the codegen in include.

cd include
python codegen.py

Then build with cmake.

cd benchmark_cpp
mkdir build
cd build
cmake ..
make

Run the reuslt

cd build
./benchmark ${.npy file} ${N}

Run python benchmark It's better to install torch_sparse and torch_scatter via a different conda environment