This is an automated email from the ASF dual-hosted git repository.
marcoabreu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 4c55d06 Enable dp4a for the CI (#9684)
4c55d06 is described below
commit 4c55d06a457daf484ae7d710a7d05cc1d73e8962
Author: Kellen Sunderland <kellen.sunderl...@gmail.com>
AuthorDate: Sun Feb 4 16:36:31 2018 +0100
Enable dp4a for the CI (#9684)
---
tests/ci_build/ci_build.sh | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/tests/ci_build/ci_build.sh b/tests/ci_build/ci_build.sh
index 794a4c5..768a2ed 100755
--- a/tests/ci_build/ci_build.sh
+++ b/tests/ci_build/ci_build.sh
@@ -168,6 +168,10 @@ echo "Running '${COMMAND[@]}' inside ${DOCKER_IMG_NAME}..."
# ${WORKDIR} is the working directory we start the container in. By default
this
# is /workspace, but for example when running cmake it is sometimes
/workspace/build.
+# Currently compiling for sm52 (g3s) and sm61(p3s). sm61 is required to test
DP4A support on p3
+# instances. In the long term we should enable sm70 for p3 instances, but
this will require
+# CUDA 9(.1) support with our base AMI drivers.
+
${DOCKER_BINARY} run --rm --pid=host \
-v ${WORKSPACE}:/workspace \
-w ${WORKDIR} \
@@ -176,7 +180,7 @@ ${DOCKER_BINARY} run --rm --pid=host \
-e "CI_BUILD_UID=$(id -u)" \
-e "CI_BUILD_GROUP=$(id -g -n)" \
-e "CI_BUILD_GID=$(id -g)" \
- -e "CUDA_ARCH=-gencode arch=compute_52,code=[sm_52,compute_52]
--fatbin-options -compress-all" \
+ -e "CUDA_ARCH=-gencode arch=compute_52,code=[sm_52,compute_52] -gencode
arch=compute_61,code=sm_61 --fatbin-options -compress-all" \
-e "MXNET_STORAGE_FALLBACK_LOG_VERBOSE=0" \
${CI_DOCKER_EXTRA_PARAMS[@]} \
${DOCKER_IMG_NAME} \
--
To stop receiving notification emails like this one, please contact
marcoab...@apache.org.
