f090b05346
* add a virtual dependency name instead of complete package name * add OneAPI components as providers of virtual packages * Revert the default of tbb --------- Co-authored-by: Nisarg Patel <nisarg.patel@lrz.de>
111 lines
4.2 KiB
Python
111 lines
4.2 KiB
Python
# Copyright 2013-2023 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 spack.package import *
|
|
|
|
|
|
@IntelOneApiPackage.update_description
|
|
class IntelOneapiDnn(IntelOneApiLibraryPackage):
|
|
"""The Intel oneAPI Deep Neural Network Library (oneDNN) helps
|
|
developers improve productivity and enhance the performance of
|
|
their deep learning frameworks. It supports key data type
|
|
formats, including 16 and 32-bit floating point, bfloat16, and
|
|
8-bit integers and implements rich operators, including
|
|
convolution, matrix multiplication, pooling, batch
|
|
normalization, activation functions, recurrent neural network
|
|
(RNN) cells, and long short-term memory (LSTM) cells.
|
|
|
|
"""
|
|
|
|
maintainers("rscohn2")
|
|
|
|
homepage = (
|
|
"https://software.intel.com/content/www/us/en/develop/tools/oneapi/components/onednn.html"
|
|
)
|
|
|
|
version(
|
|
"2023.1.0",
|
|
url="https://registrationcenter-download.intel.com/akdlm/IRC_NAS/df0fd85e-f52a-437a-8d49-be12b560607c/l_onednn_p_2023.1.0.46343_offline.sh",
|
|
sha256="0dfe16e7e81d0bf21b304e22f0cf9cb02cd4c10febddbcefea75bab2231a46d2",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2023.0.0",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/19137/l_onednn_p_2023.0.0.25399_offline.sh",
|
|
sha256="f974901132bf55ba11ce782747ba9443f38d67827bce3994775eeb86ed018869",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2022.2.1",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/19035/l_onednn_p_2022.2.1.16994_offline.sh",
|
|
sha256="2102964a36a5b58b529385706e6829456ee5225111c33dfce6326fff5175aace",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2022.2.0",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18933/l_onednn_p_2022.2.0.8750_offline.sh",
|
|
sha256="920833cd1f05f2fdafb942c96946c3925eb734d4458d52f22f2cc755133cb9e0",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2022.1.0",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18725/l_onednn_p_2022.1.0.132_offline.sh",
|
|
sha256="0b9a7efe8dd0f0b5132b353a8ee99226f75bae4bab188a453817263a0684cc93",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2022.0.2",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18476/l_onednn_p_2022.0.2.43_offline.sh",
|
|
sha256="a2a953542b4f632b51a2527d84bd76c3140a41c8085420da4237e2877c27c280",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2022.0.1",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18441/l_onednn_p_2022.0.1.26_offline.sh",
|
|
sha256="8339806300d83d2629952e6e2f2758b52f517c072a20b7b7fc5642cf1e2a5410",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2021.4.0",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18221/l_onednn_p_2021.4.0.467_offline.sh",
|
|
sha256="30cc601467f6a94b3d7e14f4639faf0b12fdf6d98df148b07acdb4dfdfb971db",
|
|
expand=False,
|
|
)
|
|
version(
|
|
"2021.3.0",
|
|
url="https://registrationcenter-download.intel.com/akdlm/irc_nas/17923/l_onednn_p_2021.3.0.344_offline.sh",
|
|
sha256="1521f6cbffcf9ce0c7b5dfcf1a2546a4a0c8d8abc99f3011709039aaa9e0859a",
|
|
expand=False,
|
|
)
|
|
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("tbb")
|
|
|
|
@property
|
|
def component_dir(self):
|
|
return "dnnl"
|
|
|
|
@property
|
|
def headers(self):
|
|
include_path = join_path(self.component_prefix, "cpu_dpcpp_gpu_dpcpp", "include")
|
|
return find_headers("dnnl", include_path)
|
|
|
|
@property
|
|
def libs(self):
|
|
lib_path = join_path(self.component_prefix, "cpu_dpcpp_gpu_dpcpp", "lib")
|
|
return find_libraries(["libdnnl", "libmkldnn"], root=lib_path, shared=True)
|