diff --git a/hpctestlib/microbenchmarks/mpi/osu.py b/hpctestlib/microbenchmarks/mpi/osu.py index 7a94dc4c1a..c405b5693d 100644 --- a/hpctestlib/microbenchmarks/mpi/osu.py +++ b/hpctestlib/microbenchmarks/mpi/osu.py @@ -145,7 +145,8 @@ def setup_per_benchmark(self): self.executable = bench.split('.')[-1] self.executable_opts = ['-m', f'{self.message_size}', '-x', f'{self.num_warmup_iters}', - '-i', f'{self.num_iters}'] + '-i', f'{self.num_iters}', '-c'] + if self.device_buffers != 'cpu': self.executable_opts += ['-d', self.device_buffers] @@ -161,7 +162,7 @@ def setup_per_benchmark(self): @sanity_function def validate_test(self): - return sn.assert_found(rf'^{self.message_size}', self.stdout) + return sn.assert_found(rf'^{self.message_size}.*Pass', self.stdout) @deferrable def _extract_metric(self):