diff options
-rw-r--r-- | var/spack/repos/builtin/packages/intel-oneapi-dnn/package.py | 10 |
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) |