[tvm] branch main updated: [ci][docker gpu] Install dnnl in docker GPU. (#11744)

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 1b8f3b54c6 [ci][docker gpu] Install dnnl in docker GPU. (#11744)
1b8f3b54c6 is described below

commit 1b8f3b54c6da0bf25427a8bd9a049d87318a1e66
Author: Hua Jiang <hu...@xilinx.com>
AuthorDate: Thu Jun 16 22:28:31 2022 -0700

    [ci][docker gpu] Install dnnl in docker GPU. (#11744)
    
    BYOC related tutorial may use dnnl  and such tutorial run at docker gpu
    which need to install dnnl to prepare the environment.
---
 docker/Dockerfile.ci_gpu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 22c372cc70..f04d8515b8 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -139,6 +139,10 @@ COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh
 RUN bash /install/ubuntu_install_sccache.sh
 ENV PATH /opt/sccache:$PATH
 
+# dnnl
+COPY install/ubuntu_install_dnnl.sh /install/ubuntu_install_dnnl.sh
+RUN bash /install/ubuntu_install_dnnl.sh
+
 # Environment variables
 ENV PATH=/usr/local/nvidia/bin:${PATH}
 ENV PATH=/usr/local/cuda/bin:${PATH}