Skip to content

GitLab

Unverified Commit 9e26c8e3 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] upgrade CUDA to 17 and update CI gpu image to torch2.0+cu118 (#6505)

parent 864fd4f7
Hide whitespace changes
Inline Side-by-side
Showing with 19 additions and 20 deletions
CMakeLists.txt
View file @ 9e26c8e3
cmake_minimum_required(VERSION 3.8)
cmake_minimum_required(VERSION 3.18)
########################################
# Borrowed and adapted from TVM project
########################################
......
Jenkinsfile
View file @ 9e26c8e3
......@@ -317,7 +317,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711"
image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root"
alwaysPull true
}
......@@ -336,7 +336,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-gpu:cu116_v230711"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root"
alwaysPull true
}
......@@ -391,7 +391,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711"
image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root"
alwaysPull true
}
......@@ -410,7 +410,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v230711"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root --runtime nvidia"
alwaysPull true
}
......@@ -463,7 +463,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v230711"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root --runtime nvidia"
alwaysPull true
}
......@@ -488,7 +488,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711"
image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root --shm-size=4gb"
alwaysPull true
}
......@@ -541,7 +541,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v230711"
image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root --runtime nvidia --shm-size=8gb"
alwaysPull true
}
......@@ -570,7 +570,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711"
image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root --shm-size=4gb"
alwaysPull true
}
......@@ -618,7 +618,7 @@ pipeline {
agent {
docker {
label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711"
image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root"
alwaysPull true
}
......
docker/install/conda_env/mxnet_cpu.yml
View file @ 9e26c8e3
......@@ -7,7 +7,7 @@ dependencies:
- pytest
- nose
- numpy
- cython
- cython==0.29
- scipy
- networkx
- matplotlib
......
docker/install/conda_env/mxnet_gpu.yml
View file @ 9e26c8e3
......@@ -7,7 +7,7 @@ dependencies:
- pytest
- nose
- numpy
- cython
- cython==0.29
- scipy
- networkx
- matplotlib
......
docker/install/conda_env/tensorflow_cpu.yml
View file @ 9e26c8e3
......@@ -7,7 +7,7 @@ dependencies:
- pytest
- nose
- numpy
- cython
- cython==0.29
- scipy
- networkx
- matplotlib
......
docker/install/conda_env/tensorflow_gpu.yml
View file @ 9e26c8e3
......@@ -7,7 +7,7 @@ dependencies:
- pytest
- nose
- numpy
- cython
- cython==0.29
- scipy
- networkx
- matplotlib
......
docker/install/ubuntu_install_build.sh
View file @ 9e26c8e3
# install cmake 3.15, cmake>=3.12 is required for CUDA 10.1
version=3.15
build=5
# Install cmake with minimum required version.
version=3.18
build=0
mkdir ~/temp
cd ~/temp
wget https://cmake.org/files/v$version/cmake-$version.$build-Linux-x86_64.sh
......
graphbolt/CMakeLists.txt
View file @ 9e26c8e3
cmake_minimum_required(VERSION 3.5)
cmake_minimum_required(VERSION 3.18)
project(graphbolt C CXX)
set (CMAKE_CXX_STANDARD 17)
......@@ -57,9 +57,8 @@ target_include_directories(${LIB_GRAPHBOLT_NAME} PRIVATE ${BOLT_DIR}
"../third_party/pcg/include")
target_link_libraries(${LIB_GRAPHBOLT_NAME} "${TORCH_LIBRARIES}")
# TODO: upgrade to 17 for consistency with CXX standard once our linux CI supports it.
if(USE_CUDA)
set_target_properties(${LIB_GRAPHBOLT_NAME} PROPERTIES CUDA_STANDARD 14)
set_target_properties(${LIB_GRAPHBOLT_NAME} PROPERTIES CUDA_STANDARD 17)
endif()
# The Torch CMake configuration only sets up the path for the MKL library when
......
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