From c03d4688e42b30bb7a9299d1a523bcf7ac55c080 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sun, 10 May 2020 19:40:00 -0500 Subject: PyTorch: fix master and mkldnn builds (#16560) --- var/spack/repos/builtin/packages/py-torch/package.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 024e0d95b2..ce69a5581c 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -247,7 +247,7 @@ class PyTorch(PythonPackage, CudaPackage): enable_or_disable('mkldnn') if '@0.4:0.4.1+mkldnn' in self.spec: - env.set('MKLDNN_HOME', self.spec['intel-mkl-dnn'].prefix) + env.set('MKLDNN_HOME', self.spec['onednn'].prefix) enable_or_disable('nnpack') enable_or_disable('qnnpack') @@ -270,8 +270,9 @@ class PyTorch(PythonPackage, CudaPackage): enable_or_disable('lmdb', newer=True) enable_or_disable('binary', keyword='BUILD', newer=True) - env.set('PYTORCH_BUILD_VERSION', self.version) - env.set('PYTORCH_BUILD_NUMBER', 0) + if not self.spec.satisfies('@master'): + env.set('PYTORCH_BUILD_VERSION', self.version) + env.set('PYTORCH_BUILD_NUMBER', 0) # BLAS to be used by Caffe2 if '^mkl' in self.spec: -- cgit v1.2.3-70-g09d2