Skip to content

Commit 0ff8fea

Browse files
authored
Merge pull request #896 from teojgo/test/tesorflow_pe19.06
[test] Update Tensorflow checks for PE19.06
2 parents 44c5783 + e632051 commit 0ff8fea

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

cscs-checks/apps/tensorflow/tensorflow_check.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
class TensorFlowBaseTest(rfm.RunOnlyRegressionTest):
66
def __init__(self, model_name):
7-
super().__init__()
87
self.name = 'tensorflow_%s_check' % model_name
98
self.descr = 'Tensorflow official %s test' % model_name
109
self.valid_systems = ['daint:gpu', 'dom:gpu']
@@ -14,11 +13,14 @@ def __init__(self, model_name):
1413
self.tags = {'production'}
1514
self.num_tasks = 1
1615
self.num_gpus_per_node = 1
17-
self.modules = ['TensorFlow/1.12.0-CrayGNU-19.03-cuda-10.0-python3']
16+
tf_version = '1.14.0'
17+
cuda_version = '10.1.168'
18+
self.modules = ['TensorFlow/%s-CrayGNU-19.06-cuda-%s-python3' %
19+
(tf_version, cuda_version)]
1820

1921
# Checkout to the branch corresponding to the module version of
2022
# TensorFlow
21-
self.pre_run = ['git checkout r1.12.0']
23+
self.pre_run = ['git checkout r%s' % tf_version]
2224
self.variables = {'PYTHONPATH': '$PYTHONPATH:.'}
2325

2426

0 commit comments

Comments
 (0)