Set MAX_JOBS and fix typo

See easybuilders/easybuild-easyconfigs#10772
Flamefire · Jul 23, 2020 · 919d22b · 919d22b
1 parent b7761e1
commit 919d22b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/easybuild/easyblocks/p/pytorch.py b/easybuild/easyblocks/p/pytorch.py
@@ -75,6 +75,8 @@ def configure_step(self):
         super(EB_PyTorch, self).configure_step()
 
         options = ['PYTORCH_BUILD_VERSION=' + self.version, 'PYTORCH_BUILD_NUMBER=1']
+        # Restrict parallelism
+        options.append('MAX_JOBS=%(parallel)s')
 
         # BLAS Interface
         if get_software_root('imkl'):
@@ -94,7 +96,7 @@ def configure_step(self):
         #               'Eigen', 'GoogleBenchmark'
         if 'protobuf' in dependency_names:
             options.append('BUILD_CUSTOM_PROTOBUF=0')
-        # Always us Infiniband
+        # Always use Infiniband
         options.append('USE_IBVERBS=1')
 
         cudnn_root = get_software_root('cuDNN')