From d5d6344b3cbf1dbaad81dd2b885fde85c00bcfad Mon Sep 17 00:00:00 2001
From: "Kurt A. O'Hearn" <ohearnku@msu.edu>
Date: Mon, 24 May 2021 16:38:11 -0400
Subject: [PATCH] Tools: add keyword for setting number of CUDA streams
(gpu_streams).
---
tools/run_sim.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tools/run_sim.py b/tools/run_sim.py
index a03b44e3..c62f91aa 100644
--- a/tools/run_sim.py
+++ b/tools/run_sim.py
@@ -28,6 +28,8 @@ class TestCase():
r'(?P<key>\bdt\b\s+)\S+(?P<comment>.*)', r'\g<key>%s\g<comment>' % x, l), \
'gpus_per_node': lambda l, x: sub(
r'(?P<key>\bgpus_per_node\b\s+)\S+(?P<comment>.*)', r'\g<key>%s\g<comment>' % x, l), \
+ 'gpu_streams': lambda l, x: sub(
+ r'(?P<key>\bgpu_streams\b\s+)\S+(?P<comment>.*)', r'\g<key>%s\g<comment>' % x, l), \
'proc_by_dim': lambda l, x: sub(
r'(?P<key>\bproc_by_dim\b\s+)\S+\s+\S+\s+\S+(?P<comment>.*)',
r'\g<key>{0} {1} {2}\g<comment>'.format(*(x.split(':'))), l), \
@@ -901,6 +903,7 @@ if __name__ == '__main__':
'nsteps': ['20'],
'dt': ['0.25'],
'gpus_per_node': ['1'],
+ 'gpu_streams': ['5'],
'proc_by_dim': ['1:1:1'],
'periodic_boundaries': ['1'],
'reposition_atoms': ['0'],
--
GitLab