Kernel Benchmarking
September 19, 2023 · View on GitHub
1. Installing Sputnik
cd $FlashLLM_HOME/third_party/sputnik
mkdir build && cd build
cmake .. -DCMAKE_BUILD_TYPE=Release -DBUILD_TEST=OFF -DBUILD_BENCHMARK=OFF -DCUDA_ARCHS=”80”
make -j12
2. Installing cuSPARSELT
Following this link.
3. Building spmm_test
cd $FlashLLM_HOME/kernel_benchmark
source test_env
make
4. Benchmarking
./benchmark.sh