1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from sys import platform
from spack import *
class IntelOneapiDnn(IntelOneApiLibraryPackage):
"""Intel oneAPI DNN."""
maintainers = ['rscohn2', 'danvev']
homepage = 'https://software.intel.com/content/www/us/en/develop/tools/oneapi/components/onednn.html'
if platform == 'linux':
version('2021.2.0',
url='https://registrationcenter-download.intel.com/akdlm/irc_nas/17751/l_onednn_p_2021.2.0.228_offline.sh',
sha256='62121a3355298211a124ff4e71c42fc172bf1061019be6c6120830a1a502aa88',
expand=False)
version('2021.1.1',
url='https://registrationcenter-download.intel.com/akdlm/irc_nas/17385/l_onednn_p_2021.1.1.55_offline.sh',
sha256='24002c57bb8931a74057a471a5859d275516c331fd8420bee4cae90989e77dc3',
expand=False)
depends_on('intel-oneapi-tbb')
@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)
|
