From 8d01823296eae8efd89f3136a43778bdcc607c3c Mon Sep 17 00:00:00 2001 From: AJ Schmidt Date: Tue, 24 Nov 2020 15:10:26 -0500 Subject: [PATCH] update cugraph build command --- generated-dockerfiles/centos7-devel.Dockerfile | 2 +- generated-dockerfiles/ubuntu18.04-devel.Dockerfile | 2 +- templates/Devel.dockerfile.j2 | 3 +++ 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/generated-dockerfiles/centos7-devel.Dockerfile b/generated-dockerfiles/centos7-devel.Dockerfile index e50889b3..cb6dbd76 100644 --- a/generated-dockerfiles/centos7-devel.Dockerfile +++ b/generated-dockerfiles/centos7-devel.Dockerfile @@ -180,7 +180,7 @@ RUN cd ${RAPIDS_DIR}/cuml && \ RUN cd ${RAPIDS_DIR}/cugraph && \ source activate rapids && \ ccache -s && \ - ./build.sh + ./build.sh --allgpuarch RUN cd ${RAPIDS_DIR}/xgboost && \ source activate rapids && \ diff --git a/generated-dockerfiles/ubuntu18.04-devel.Dockerfile b/generated-dockerfiles/ubuntu18.04-devel.Dockerfile index 7dec8100..4714e4c3 100644 --- a/generated-dockerfiles/ubuntu18.04-devel.Dockerfile +++ b/generated-dockerfiles/ubuntu18.04-devel.Dockerfile @@ -180,7 +180,7 @@ RUN cd ${RAPIDS_DIR}/cuml && \ RUN cd ${RAPIDS_DIR}/cugraph && \ source activate rapids && \ ccache -s && \ - ./build.sh + ./build.sh --allgpuarch RUN cd ${RAPIDS_DIR}/xgboost && \ source activate rapids && \ diff --git a/templates/Devel.dockerfile.j2 b/templates/Devel.dockerfile.j2 index 1f1a7453..374c06c4 100644 --- a/templates/Devel.dockerfile.j2 +++ b/templates/Devel.dockerfile.j2 @@ -149,6 +149,9 @@ RUN cd ${RAPIDS_DIR}/{{ lib.name }} && \ cd ../python-package && python setup.py install; \ fi + {% elif lib.name == "cugraph" %} + ./build.sh --allgpuarch + {% elif lib.name == "cuml" %} ./build.sh --allgpuarch --buildgtest libcuml cuml prims