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

Revert "[Feature][REVIEW] Enable DGL cugaph nightly CI (#4525)" (#4563)

This reverts commit ec171c64.
parent ec171c64
...@@ -296,9 +296,9 @@ pipeline { ...@@ -296,9 +296,9 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" label "linux-cpu-node"
image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.11.0-rapids22.10" image "nvcr.io/nvidia/pytorch:22.04-py3"
args "-u root" args "-u root"
alwaysPull True alwaysPull false
} }
} }
steps { steps {
...@@ -523,9 +523,9 @@ pipeline { ...@@ -523,9 +523,9 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" label "linux-gpu-node"
image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.11.0-rapids22.10" image "nvcr.io/nvidia/pytorch:22.04-py3"
args "--runtime nvidia --shm-size=8gb" args "--runtime nvidia --shm-size=8gb"
alwaysPull True alwaysPull false
} }
} }
stages { stages {
......
...@@ -7,15 +7,12 @@ if [ $# -ne 1 ]; then ...@@ -7,15 +7,12 @@ if [ $# -ne 1 ]; then
exit -1 exit -1
fi fi
CMAKE_VARS="-DBUILD_CPP_TEST=ON -DUSE_OPENMP=ON" CMAKE_VARS="-DBUILD_CPP_TEST=ON -DUSE_OPENMP=ON -DBUILD_TORCH=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
# We do not build pytorch for cugraph because currently building CMAKE_VARS="$CMAKE_VARS -DTORCH_PYTHON_INTERPS=/opt/conda/envs/pytorch-ci/bin/python"
# 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