From fb644b08eab19aaae14ae136a08eb64d89ac4fb4 Mon Sep 17 00:00:00 2001 From: Theofilos Manitaras Date: Wed, 21 Aug 2019 13:25:54 +0200 Subject: [PATCH] Update tensorflow/horovod tests for latest modules --- cscs-checks/apps/tensorflow/tensorflow_check.py | 5 ++++- cscs-checks/apps/tensorflow/tf_horovod_check.py | 5 ++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/cscs-checks/apps/tensorflow/tensorflow_check.py b/cscs-checks/apps/tensorflow/tensorflow_check.py index c79f8ad3dd..05a06b4ad0 100644 --- a/cscs-checks/apps/tensorflow/tensorflow_check.py +++ b/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:.'} diff --git a/cscs-checks/apps/tensorflow/tf_horovod_check.py b/cscs-checks/apps/tensorflow/tf_horovod_check.py index 117f0b3dd2..586de33a01 100644 --- a/cscs-checks/apps/tensorflow/tf_horovod_check.py +++ b/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