Skip to content

Commit f780ca5

Browse files
committed
run PyTorch tests sequentially
1 parent 61898ae commit f780ca5

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

eb_hooks.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -583,6 +583,12 @@ def pre_test_hook_increase_max_failed_tests_arm_PyTorch(self, *args, **kwargs):
583583
"""
584584
if self.name == 'PyTorch' and self.version == '2.1.2' and get_cpu_architecture() == AARCH64:
585585
self.cfg['max_failed_tests'] = 10
586+
if 'cudaver' in self.cfg.template_values and self.cfg.template_values['cudaver'] == '12.1.1':
587+
_cudaver = self.cfg.template_values['cudaver']
588+
_runtest = self.cfg['runtest']
589+
self.cfg['runtest'] = _runtest.replace(
590+
'PYTHONUNBUFFERED',
591+
'PYTORCH_TEST_RUN_EVERYTHING_IN_SERIAL=1 PYTHONUNBUFFERED')
586592

587593

588594
def pre_single_extension_hook(ext, *args, **kwargs):

0 commit comments

Comments
 (0)