From 6684eea0ccccd43fd3599730f9ef0e962d46857b Mon Sep 17 00:00:00 2001
From: huajsj <huaj@xilinx.com>
Date: Wed, 15 Jun 2022 19:07:06 -0700
Subject: [PATCH] [ci][docker gpu] Install dnnl in docker GPU.

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 e0d1997de729..49b3557dbb89 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -137,6 +137,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}