From 3be438aca282c7c77b83c322a431947014b8cf87 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 30 Jun 2020 19:18:06 -0700
Subject: [PATCH]  Add horovod tests

PiperOrigin-RevId: 319145206
Change-Id: I44f186249e609caa570eef7d48b53da83214eb44
---
 .../tools/ci_build/horovod/gpu/nightly.sh     | 22 ++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/tensorflow/tools/ci_build/horovod/gpu/nightly.sh b/tensorflow/tools/ci_build/horovod/gpu/nightly.sh
index 50f5f49cfad..c85aa18b8fa 100644
--- a/tensorflow/tools/ci_build/horovod/gpu/nightly.sh
+++ b/tensorflow/tools/ci_build/horovod/gpu/nightly.sh
@@ -35,12 +35,32 @@ fi
 wget https://download.open-mpi.org/release/open-mpi/v4.0/openmpi-4.0.4.tar.gz
 tar xvf openmpi-4.0.4.tar.gz
 
-cd openmpi
+# Install gcc.
+sudo apt install --assume-yes build-essential
+
+gcc --version
+
+cd openmpi-4.0.4
 ./configure
 
 # Install open-mpi.
 sudo make all install
+export LD_LIBRARY_PATH=/usr/local/lib/openmpi
 sudo ldconfig
 
+# Install Horovod.
+cd ..
+pip3 install horovod tensorflow
 
+# Install tests.
+git clone https://github.com/DEKHTIARJonathan/TF_HVD_Stability_Test.git
 
+# Install pytest.
+pip3 install -U pytest
+
+# Install requirements.
+cd TF_HVD_Stability_Test
+pip3 install -r requirements.txt
+
+# Run the tests.
+python3 -m pytest