summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/intel-oneapi-dnn/package.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/intel-oneapi-dnn/package.py b/var/spack/repos/builtin/packages/intel-oneapi-dnn/package.py
index 499a98d850..4808303083 100644
--- a/var/spack/repos/builtin/packages/intel-oneapi-dnn/package.py
+++ b/var/spack/repos/builtin/packages/intel-oneapi-dnn/package.py
@@ -31,3 +31,13 @@ class IntelOneapiDnn(IntelOneApiLibraryPackage):
@property
def component_dir(self):
return 'dnnl'
+
+ @property
+ def headers(self):
+ include_path = join_path(self.component_path, 'cpu_dpcpp_gpu_dpcpp', 'include')
+ return find_headers('dnnl', include_path)
+
+ @property
+ def libs(self):
+ lib_path = join_path(self.component_path, 'cpu_dpcpp_gpu_dpcpp', 'lib')
+ return find_libraries(['libdnnl', 'libmkldnn'], root=lib_path, shared=True)