Skip to content

GitLab

Unverified Commit 65e6b04d authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[Build] enable USE_AVX in CI and regression tests (#4004)

parent 4177f729
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
benchmarks/scripts/build_dgl_asv.sh
View file @ 65e6b04d
...@@ -16,6 +16,10 @@ if [[ $DEVICE == "cpu" ]]; then ...@@ -16,6 +16,10 @@ if [[ $DEVICE == "cpu" ]]; then
else else
CMAKE_VARS="-DUSE_CUDA=ON" CMAKE_VARS="-DUSE_CUDA=ON"
fi fi
arch=`uname -m`
if [[ $arch == *"x86"* ]]; then
CMAKE_VARS="-DUSE_AVX=ON $CMAKE_VARS"
fi
mkdir -p build mkdir -p build
pushd build pushd build
cmake -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda -DBUILD_TORCH=ON $CMAKE_VARS .. cmake -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda -DBUILD_TORCH=ON $CMAKE_VARS ..
......
tests/scripts/build_dgl.bat
View file @ 65e6b04d
...@@ -13,7 +13,7 @@ SET TEMP=%WORKSPACE%\tmp ...@@ -13,7 +13,7 @@ SET TEMP=%WORKSPACE%\tmp
SET TMPDIR=%WORKSPACE%\tmp SET TMPDIR=%WORKSPACE%\tmp
PUSHD build PUSHD build
cmake -DCMAKE_CXX_FLAGS="/DDGL_EXPORTS" -DUSE_OPENMP=ON -DBUILD_TORCH=ON -Dgtest_force_shared_crt=ON -DDMLC_FORCE_SHARED_CRT=ON -DBUILD_CPP_TEST=1 -DCMAKE_CONFIGURATION_TYPES="Release" .. -G "Visual Studio 16 2019" || EXIT /B 1 cmake -DCMAKE_CXX_FLAGS="/DDGL_EXPORTS" -DUSE_AVX=ON -DUSE_OPENMP=ON -DBUILD_TORCH=ON -Dgtest_force_shared_crt=ON -DDMLC_FORCE_SHARED_CRT=ON -DBUILD_CPP_TEST=1 -DCMAKE_CONFIGURATION_TYPES="Release" .. -G "Visual Studio 16 2019" || EXIT /B 1
msbuild dgl.sln /m /nr:false || EXIT /B 1 msbuild dgl.sln /m /nr:false || EXIT /B 1
COPY /Y Release\runUnitTests.exe . COPY /Y Release\runUnitTests.exe .
POPD POPD
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment