[Relay] Parser CI dependencies and build rules (apache#1965)

luckyqsz · Oct 23, 2018 · 975d0d4 · 975d0d4
1 parent 81db03a
commit 975d0d4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
@@ -69,6 +69,9 @@ RUN bash /install/ubuntu_install_tensorflow.sh
 COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
 RUN bash /install/ubuntu_install_redis.sh
 
+COPY install/ubuntu_install_antlr.sh /install/ubuntu_install_antlr.sh
+RUN bash /install/ubuntu_install_antlr.sh
+
 # Environment variables
 ENV PATH=/usr/local/nvidia/bin:${PATH}
 ENV PATH=/usr/local/cuda/bin:${PATH}

diff --git a/docker/install/ubuntu_install_antlr.sh b/docker/install/ubuntu_install_antlr.sh
@@ -0,0 +1,5 @@
+cd /usr/local/lib
+wget https://www.antlr.org/download/antlr-4.7.1-complete.jar
+cd -
+
+alias antlr4='java -jar /usr/local/lib/antlr-4.7.1-complete.jar'
diff --git a/docker/install/ubuntu_install_python_package.sh b/docker/install/ubuntu_install_python_package.sh
@@ -1,3 +1,3 @@
 # install libraries for python package on ubuntu
-pip2 install nose pylint numpy nose-timer cython decorator scipy tornado
+pip2 install nose pylint numpy nose-timer cython decorator scipy tornado typing antlr4-python2-runtime
 pip3 install nose pylint numpy nose-timer cython decorator scipy tornado typed_ast pytest mypy orderedset antlr4-python3-runtime