From f780ca552270086f34798c4500978be453fe4ef8 Mon Sep 17 00:00:00 2001 From: Thomas Roeblitz Date: Wed, 29 May 2024 23:11:47 +0200 Subject: [PATCH] run PyTorch tests sequentially --- eb_hooks.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/eb_hooks.py b/eb_hooks.py index f01e1ca55c..e4a957acdc 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -583,6 +583,12 @@ def pre_test_hook_increase_max_failed_tests_arm_PyTorch(self, *args, **kwargs): """ if self.name == 'PyTorch' and self.version == '2.1.2' and get_cpu_architecture() == AARCH64: self.cfg['max_failed_tests'] = 10 + if 'cudaver' in self.cfg.template_values and self.cfg.template_values['cudaver'] == '12.1.1': + _cudaver = self.cfg.template_values['cudaver'] + _runtest = self.cfg['runtest'] + self.cfg['runtest'] = _runtest.replace( + 'PYTHONUNBUFFERED', + 'PYTORCH_TEST_RUN_EVERYTHING_IN_SERIAL=1 PYTHONUNBUFFERED') def pre_single_extension_hook(ext, *args, **kwargs):