"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8e963d1c2a2b6ebf2880675516809bd9a39359a4"
Unverified Commit be114093 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[CI] Enable DGL cugaph nightly CI (#4573)

* [CI] Enable DGL cugaph nightly CI

* fix build issue

* remove runtime
parent dc258603
...@@ -298,9 +298,9 @@ pipeline { ...@@ -298,9 +298,9 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" 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" args "-u root"
alwaysPull false alwaysPull true
} }
} }
steps { steps {
...@@ -525,9 +525,9 @@ pipeline { ...@@ -525,9 +525,9 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" 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" args "--runtime nvidia --shm-size=8gb"
alwaysPull false alwaysPull true
} }
} }
stages { stages {
......
...@@ -7,12 +7,15 @@ if [ $# -ne 1 ]; then ...@@ -7,12 +7,15 @@ if [ $# -ne 1 ]; then
exit -1 exit -1
fi 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. # 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 # The value here is for CI. Replace it with your own or comment this whole
# statement for default Python interpreter. # statement for default Python interpreter.
if [ "$1" != "cugraph" ]; then 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 fi
#This is implemented to detect underlying architecture and enable arch specific optimization. #This is implemented to detect underlying architecture and enable arch specific optimization.
......
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