diff --git a/easybuild/easyconfigs/p/PyTorch/PyTorch-1.12.1-foss-2021b.eb b/easybuild/easyconfigs/p/PyTorch/PyTorch-1.12.1-foss-2021b.eb new file mode 100644 index 00000000000..7c58b58e5fd --- /dev/null +++ b/easybuild/easyconfigs/p/PyTorch/PyTorch-1.12.1-foss-2021b.eb @@ -0,0 +1,138 @@ +name = 'PyTorch' +version = '1.12.1' + +homepage = 'https://pytorch.org/' +description = """Tensors and Dynamic neural networks in Python with strong GPU acceleration. +PyTorch is a deep learning framework that puts Python first.""" + +toolchain = {'name': 'foss', 'version': '2021b'} + +source_urls = [GITHUB_RELEASE] +sources = ['%(namelower)s-v%(version)s.tar.gz'] + +patches = [ + 'PyTorch-1.7.0_avoid-nan-in-test-torch.patch', + 'PyTorch-1.7.0_disable-dev-shm-test.patch', + 'PyTorch-1.10.0_fix-kineto-crash.patch', + 'PyTorch-1.10.0_fix-test-dataloader-fixed-affinity.patch', + 'PyTorch-1.10.0_fix-test-model_dump.patch', + 'PyTorch-1.10.0_fix-vsx-vector-functions.patch', + 'PyTorch-1.10.0_skip-nnapi-test-without-qnnpack.patch', + 'PyTorch-1.11.0_fix-fsdp-fp16-test.patch', + 'PyTorch-1.11.0_fix-test_utils.patch', + 'PyTorch-1.11.0_increase_c10d_gloo_timeout.patch', + 'PyTorch-1.11.0_increase-distributed-test-timeout.patch', + 'PyTorch-1.11.0_install-vsx-vec-headers.patch', + 'PyTorch-1.12.1_fix-cuda-gcc-version-check.patch', + 'PyTorch-1.12.1_fix-skip-decorators.patch', + 'PyTorch-1.12.1_fix-test_cpp_extensions_jit.patch', + 'PyTorch-1.12.1_fix-test_wishart_log_prob.patch', + 'PyTorch-1.12.1_fix-TestCudaFuser.test_unary_ops.patch', + 'PyTorch-1.12.1_fix-TestTorch.test_to.patch', + 'PyTorch-1.12.1_fix-use-after-free-in-tensorpipe-agent.patch', + 'PyTorch-1.12.1_fix-vsx-vector-funcs.patch', + 'PyTorch-1.12.1_fix-vsx-loadu.patch', + 'PyTorch-1.12.1_increase-test-adadelta-tolerance.patch', + 'PyTorch-1.12.1_increase-tolerance-test_ops.patch', + 'PyTorch-1.12.1_no-cuda-stubs-rpath.patch', + 'PyTorch-1.12.1_python-3.10-annotation-fix.patch', + 'PyTorch-1.12.1_python-3.10-compat.patch', + 'PyTorch-1.12.1_remove-flaky-test-in-testnn.patch', + 'PyTorch-1.12.1_skip-ao-sparsity-test-without-fbgemm.patch', + 'PyTorch-1.12.1_skip-failing-grad-test.patch', + 'PyTorch-1.12.1_skip-test_round_robin.patch', +] +checksums = [ + '031c71073db73da732b5d01710220564ce6dd88d812ba053f0cc94296401eccb', # pytorch-v1.12.1.tar.gz + 'b899aa94d9e60f11ee75a706563312ccefa9cf432756c470caa8e623991c8f18', # PyTorch-1.7.0_avoid-nan-in-test-torch.patch + '622cb1eaeadc06e13128a862d9946bcc1f1edd3d02b259c56a9aecc4d5406b8a', # PyTorch-1.7.0_disable-dev-shm-test.patch + # PyTorch-1.10.0_fix-kineto-crash.patch + 'dc467333b28162149af8f675929d8c6bf219f23230bfc0d39af02ba4f6f882eb', + # PyTorch-1.10.0_fix-test-dataloader-fixed-affinity.patch + '313dca681f45ce3bc7c4557fdcdcbe0b77216d2c708fa30a2ec0e22c44876707', + # PyTorch-1.10.0_fix-test-model_dump.patch + '339148ae1a028cda6e750ac93fa38a599f66c7abe26586c9219f1a206ea14557', + # PyTorch-1.10.0_fix-vsx-vector-functions.patch + '7bef5f96cb83b2d655d2f76dd7468a171d446f0b3e06da2232ec7f886484d312', + # PyTorch-1.10.0_skip-nnapi-test-without-qnnpack.patch + '34ba476a7bcddec323bf9eca083cb4623d0f569d081aa3add3769c24f22849d2', + 'bb1c4e6d6fd4b0cf57ff8b824c797331b533bb1ffc63f5db0bae3aee10c3dc13', # PyTorch-1.11.0_fix-fsdp-fp16-test.patch + '4f7e25c4e2eb7094f92607df74488c6a4a35849fabf05fcf6c3655fa3f44a861', # PyTorch-1.11.0_fix-test_utils.patch + # PyTorch-1.11.0_increase_c10d_gloo_timeout.patch + '20cd4a8663f74ab326fdb032b926bf5c7e94d9750c515ab9050927ba00cf1953', + # PyTorch-1.11.0_increase-distributed-test-timeout.patch + '087ad20163a1291773ae3457569b80523080eb3731e210946459b2333a919f3f', + 'f2e6b9625733d9a471bb75e1ea20e28814cf1380b4f9089aa838ee35ddecf07d', # PyTorch-1.11.0_install-vsx-vec-headers.patch + # PyTorch-1.12.1_fix-cuda-gcc-version-check.patch + 'a650f4576f06c749f244cada52ff9c02499fa8f182019129488db3845e0756ab', + 'e3ca6e42b2fa592ea095939fb59ab875668a058479407db3f3684cc5c6f4146c', # PyTorch-1.12.1_fix-skip-decorators.patch + # PyTorch-1.12.1_fix-test_cpp_extensions_jit.patch + '1efc9850c431d702e9117d4766277d3f88c5c8b3870997c9974971bce7f2ab83', + # PyTorch-1.12.1_fix-test_wishart_log_prob.patch + 'cf475ae6e6234b96c8d1bf917597c5176c94b3ccd940b72f2e1cd0c979580f45', + # PyTorch-1.12.1_fix-TestCudaFuser.test_unary_ops.patch + '8e6e844c6b0541e0c8115911ee1a9d548613254b36dfbdada202fd723fc26aa2', + '75f27987c3f25c501e719bd2b1c70a029ae0ee28514a97fe447516aee02b1535', # PyTorch-1.12.1_fix-TestTorch.test_to.patch + # PyTorch-1.12.1_fix-use-after-free-in-tensorpipe-agent.patch + '0bd7e88b92c4c6f0fecf01746009858ba19f2df68b10b88c41485328a531875d', + 'caccbf60f62eac313896c1eaec78b08f5d0fdfcb907079087490bb13d1561aa2', # PyTorch-1.12.1_fix-vsx-vector-funcs.patch + '8bfe3c94ada1dd1f7974a1261a8b576fb7ae944050fa1c7830fca033831123b2', # PyTorch-1.12.1_fix-vsx-loadu.patch + # PyTorch-1.12.1_increase-test-adadelta-tolerance.patch + '944ed1af5ad4bbe20cbb042764a88dad1eef6cd33218617cf3d4cd90c6764695', + # PyTorch-1.12.1_increase-tolerance-test_ops.patch + '1c1fa520801e2ee5faf56a3d6dc96321e7c11664fd16bffd7c6ee437e68357fb', + '2905826ca713752b47c84e4ec8b177c90cbd91fca498ba2ba546f495c4cf70a6', # PyTorch-1.12.1_no-cuda-stubs-rpath.patch + # PyTorch-1.12.1_python-3.10-annotation-fix.patch + '11e168fd429d9e156fc79dd806b08125f3640651ad9998abd810446b2ed0c2d7', + '81402420a878b40f824778f0333fbec6504325a6a1b06a22749c4cac3eaccf67', # PyTorch-1.12.1_python-3.10-compat.patch + # PyTorch-1.12.1_remove-flaky-test-in-testnn.patch + 'e81b678e354dd137c0d6d974605cdedbf672096fdbdf567c347bc2fbfc73471d', + # PyTorch-1.12.1_skip-ao-sparsity-test-without-fbgemm.patch + 'edd464ec8c37b44c07a72008d732604f6837f2dd61c7810c391a86ba4945ca39', + '1c89e7e67287fe6b9a95480a4178d3653b94d0ab2fe68edf227606c8ae548fdc', # PyTorch-1.12.1_skip-failing-grad-test.patch + # PyTorch-1.12.1_skip-test_round_robin.patch + '63d4849b78605aa088fdff695637d9473ea60dee603a3ff7f788690d70c55349', +] + +osdependencies = [OS_PKG_IBVERBS_DEV] + +builddependencies = [ + ('CMake', '3.22.1'), + ('hypothesis', '6.14.6'), +] + +dependencies = [ + ('Ninja', '1.10.2'), # Required for JIT compilation of C++ extensions + ('Python', '3.9.6'), + ('protobuf', '3.17.3'), + ('protobuf-python', '3.17.3'), + ('pybind11', '2.7.1'), + ('SciPy-bundle', '2021.10'), + ('typing-extensions', '3.10.0.2'), + ('PyYAML', '5.4.1'), + ('MPFR', '4.1.0'), + ('GMP', '6.2.1'), + ('numactl', '2.0.14'), + ('FFmpeg', '4.3.2'), + ('Pillow', '8.3.2'), + ('expecttest', '0.1.3'), +] + +excluded_tests = { + '': [ + # This test seems to take too long on NVIDIA Ampere at least. + 'distributed/test_distributed_spawn', + # failing on broadwell + # See https://github.com/easybuilders/easybuild-easyconfigs/issues/17615 + 'test_native_mha', + # flaky test which fails on some systems, + # see https://github.com/easybuilders/easybuild-easyconfigs/issues/17615 + 'distributed/elastic/multiprocessing/api_test', + ] +} + +runtest = 'cd test && PYTHONUNBUFFERED=1 %(python)s run_test.py --continue-through-error --verbose %(excluded_tests)s' + +tests = ['PyTorch-check-cpp-extension.py'] + +moduleclass = 'ai'