Update tensorflow/horovod tests for latest modules

Author: Theofilos Manitaras

cscs-checks/apps/tensorflow/tensorflow_check.py

@@ -20,7 +20,10 @@ def __init__(self, model_name):
 
         # Checkout to the branch corresponding to the module version of
         # TensorFlow
-        self.pre_run = ['git checkout r%s' % tf_version]
+        # FIXME: Currently the branch for Tensorflow 1.14.0 is not
+        # available, we use the one for 1.13.0
+        # self.pre_run = ['git checkout r%s' % tf_version]
+        self.pre_run = ['git checkout r1.13.0']
         self.variables = {'PYTHONPATH': '$PYTHONPATH:.'}
 
 

cscs-checks/apps/tensorflow/tf_horovod_check.py

@@ -6,13 +6,12 @@
 @rfm.parameterized_test(['small'], ['large'])
 class TensorFlowHorovodTest(rfm.RunOnlyRegressionTest):
     def __init__(self, variant):
-        super().__init__()
         self.descr = 'Distributed training with TensorFlow and Horovod'
         self.valid_systems = ['daint:gpu']
         self.valid_prog_environs = ['PrgEnv-gnu']
-        tfshortver = '1.12'
+        tfshortver = '1.14'
         self.sourcesdir = 'https://github.com/tensorflow/benchmarks'
-        self.modules = ['Horovod/0.16.0-CrayGNU-19.03-tf-%s.0' % tfshortver]
+        self.modules = ['Horovod/0.16.4-CrayGNU-19.06-tf-%s.0' % tfshortver]
         if variant == 'small':
             self.valid_systems += ['dom:gpu']
             self.num_tasks = 8