diff --git a/easystacks/software.eessi.io/2023.06/sapphire_rapids/eessi-2023.06-eb-4.9.0-2023a.yml b/easystacks/software.eessi.io/2023.06/sapphire_rapids/eessi-2023.06-eb-4.9.0-2023a.yml index 2b53315214..4e5f051d2d 100644 --- a/easystacks/software.eessi.io/2023.06/sapphire_rapids/eessi-2023.06-eb-4.9.0-2023a.yml +++ b/easystacks/software.eessi.io/2023.06/sapphire_rapids/eessi-2023.06-eb-4.9.0-2023a.yml @@ -41,3 +41,7 @@ easyconfigs: options: from-pr: 19471 include-easyblocks-from-pr: 3036 + - PyTorch-2.1.2-foss-2023a.eb: + options: + # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19573 + from-pr: 19573 diff --git a/eb_hooks.py b/eb_hooks.py index 08eb5ea62e..9f3793a0db 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -26,6 +26,7 @@ CPU_TARGET_AARCH64_GENERIC = 'aarch64/generic' CPU_TARGET_A64FX = 'aarch64/a64fx' +CPU_TARGET_SAPPHIRE_RAPIDS = 'x86_64/intel/sapphire_rapids' CPU_TARGET_ZEN4 = 'x86_64/amd/zen4' EESSI_RPATH_OVERRIDE_ATTR = 'orig_rpath_override_dirs' @@ -817,11 +818,16 @@ def pre_test_hook_ignore_failing_tests_netCDF(self, *args, **kwargs): def pre_test_hook_increase_max_failed_tests_arm_PyTorch(self, *args, **kwargs): """ - Pre-test hook for PyTorch: increase max failing tests for ARM for PyTorch 2.1.2 - See https://github.com/EESSI/software-layer/pull/444#issuecomment-1890416171 + Pre-test hook for PyTorch: increase max failing tests for ARM and Intel Sapphire Rapids for PyTorch 2.1.2 + See https://github.com/EESSI/software-layer/pull/444#issuecomment-1890416171 and + https://github.com/EESSI/software-layer/pull/875#issuecomment-2606854400 """ - if self.name == 'PyTorch' and self.version == '2.1.2' and get_cpu_architecture() == AARCH64: - self.cfg['max_failed_tests'] = 10 + cpu_target = get_eessi_envvar('EESSI_SOFTWARE_SUBDIR') + if self.name == 'PyTorch' and self.version == '2.1.2': + if get_cpu_architecture() == AARCH64: + self.cfg['max_failed_tests'] = 10 + if cpu_target == CPU_TARGET_SAPPHIRE_RAPIDS: + self.cfg['max_failed_tests'] = 4 def pre_single_extension_hook(ext, *args, **kwargs): diff --git a/eessi-2023.06-known-issues.yml b/eessi-2023.06-known-issues.yml index c5cdc68941..bdef076dcd 100644 --- a/eessi-2023.06-known-issues.yml +++ b/eessi-2023.06-known-issues.yml @@ -56,3 +56,7 @@ - SciPy-bundle-2023.11-gfbf-2023b: - issue: https://github.com/EESSI/software-layer/issues/318 - info: "2 failing tests (vs 54876 passed) in scipy test suite" +- x86_64/intel/sapphire_rapids: + - PyTorch-2.1.2-foss-2023a: + - issue: https://github.com/EESSI/software-layer/issues/461 + - info: "4 failing tests (out of 209567) on x86_64/intel/sapphire_rapids"