Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
be114093
Unverified
Commit
be114093
authored
Sep 22, 2022
by
Rhett Ying
Committed by
GitHub
Sep 22, 2022
Browse files
[CI] Enable DGL cugaph nightly CI (#4573)
* [CI] Enable DGL cugaph nightly CI * fix build issue * remove runtime
parent
dc258603
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
6 deletions
+9
-6
Jenkinsfile
Jenkinsfile
+4
-4
tests/scripts/build_dgl.sh
tests/scripts/build_dgl.sh
+5
-2
No files found.
Jenkinsfile
View file @
be114093
...
...
@@ -298,9 +298,9 @@ pipeline {
agent
{
docker
{
label
"linux-cpu-node"
image
"
nvcr.io/nvidia/pytorch:22.04-py3
"
image
"
rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.10
"
args
"-u root"
alwaysPull
fals
e
alwaysPull
tru
e
}
}
steps
{
...
...
@@ -525,9 +525,9 @@ pipeline {
agent
{
docker
{
label
"linux-gpu-node"
image
"
nvcr.io/nvidia/pytorch:22.04-py3
"
image
"
rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.10
"
args
"--runtime nvidia --shm-size=8gb"
alwaysPull
fals
e
alwaysPull
tru
e
}
}
stages
{
...
...
tests/scripts/build_dgl.sh
View file @
be114093
...
...
@@ -7,12 +7,15 @@ if [ $# -ne 1 ]; then
exit
-1
fi
CMAKE_VARS
=
"-DBUILD_CPP_TEST=ON -DUSE_OPENMP=ON
-DBUILD_TORCH=ON
"
CMAKE_VARS
=
"-DBUILD_CPP_TEST=ON -DUSE_OPENMP=ON"
# This is a semicolon-separated list of Python interpreters containing PyTorch.
# The value here is for CI. Replace it with your own or comment this whole
# statement for default Python interpreter.
if
[
"
$1
"
!=
"cugraph"
]
;
then
CMAKE_VARS
=
"
$CMAKE_VARS
-DTORCH_PYTHON_INTERPS=/opt/conda/envs/pytorch-ci/bin/python"
# We do not build pytorch for cugraph because currently building
# pytorch against all the supported cugraph versions is not supported
# See issue: https://github.com/rapidsai/cudf/issues/8510
CMAKE_VARS
=
"
$CMAKE_VARS
-DBUILD_TORCH=ON -DTORCH_PYTHON_INTERPS=/opt/conda/envs/pytorch-ci/bin/python"
fi
#This is implemented to detect underlying architecture and enable arch specific optimization.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment