dmlc--dgl
Rhett Ying
46a3fc2b5f
* [CI] updgrade pytorch version for benchmark CI * update build arguments * updage * updage * upgrade torch to 1.13 * updage docker image * update cmake args * try with cu116_torch112 * update build * update * update * update * update docker image * update * update * update * update * final update * fix continue running * update * update * update
25 行
492 B
Bash
25 行
492 B
Bash
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
DEVICE=$1
|
|
ROOT=/asv/dgl
|
|
|
|
. /opt/conda/etc/profile.d/conda.sh
|
|
|
|
conda activate base
|
|
pip install --upgrade pip
|
|
# Newer asv version like 0.5.1 has different result format,
|
|
# so we fix the version here. Or `generate_excel.py` has to be changed.
|
|
pip install asv==0.4.2
|
|
pip uninstall -y dgl
|
|
|
|
export DGL_BENCH_DEVICE=$DEVICE
|
|
echo "DGL_BENCH_DEVICE=$DGL_BENCH_DEVICE"
|
|
pushd $ROOT/benchmarks
|
|
cat asv.conf.json
|
|
asv machine --yes
|
|
asv run --launch-method=spawn -e -v
|
|
asv publish
|
|
popd
|