diff options
Diffstat (limited to 'dev-libs/oneDNN')
| -rw-r--r-- | dev-libs/oneDNN/Manifest | 3 | ||||
| -rw-r--r-- | dev-libs/oneDNN/metadata.xml | 24 | ||||
| -rw-r--r-- | dev-libs/oneDNN/oneDNN-3.3.3.ebuild | 106 |
3 files changed, 133 insertions, 0 deletions
diff --git a/dev-libs/oneDNN/Manifest b/dev-libs/oneDNN/Manifest new file mode 100644 index 000000000000..922ef95e2e53 --- /dev/null +++ b/dev-libs/oneDNN/Manifest @@ -0,0 +1,3 @@ +DIST oneDNN-3.3.3.tar.gz 11577369 BLAKE2B 85df74235d70d30d69ebb1702dcb328466d38046b37f4078b1d6880c7cbe079d095e53e5c9e2f0f77daca057cae62e661ac1bf919e8f128cf9c76a847dc0425d SHA512 bb4a6b33b3191b9e517a321dd089b7debb528dd2d4be48f52626308fe0e1dbea3fa6149fd9debe6aa23f11224ac80e4d3ee1e251d2007faf9bcc262bb895de71 +EBUILD oneDNN-3.3.3.ebuild 2938 BLAKE2B 2ad6b0f9ce0a02e4368594cd9bfa04ddfbdf17a39daed72d1021f3c1cd3e7e39330b555bf172b6be999bc514a8f9d2caba0b30d7ddee1c6f26620e4588718a1b SHA512 a77b3ef9e200adc50c007ea7d47292cd71be2fdd4e4d610914afb961097598285b96cd23c23cda16ee7de04f1d8fa3dece189f8b33970746faa88175659fe630 +MISC metadata.xml 777 BLAKE2B 33c65c97e0c8132e315aa389c395407e8985fbe1607222f2affb04534c33e4df14776d90f3293fc651d7892f703cc8eff5803d075291c979bcd828aa51ab1778 SHA512 9b0c2354f69d0e2656bba2995fb956f16d6fad65887f240b670548d244074e7fe92bff73afd379e9d9a208c39ac7da01de97463aa65c37dd51aded32228aafd3 diff --git a/dev-libs/oneDNN/metadata.xml b/dev-libs/oneDNN/metadata.xml new file mode 100644 index 000000000000..1e6efe71ca6a --- /dev/null +++ b/dev-libs/oneDNN/metadata.xml @@ -0,0 +1,24 @@ +<?xml version="1.0" encoding="UTF-8"?> +<!DOCTYPE pkgmetadata SYSTEM "https://www.gentoo.org/dtd/metadata.dtd"> +<pkgmetadata> + <maintainer type="person"> + <email>tupone@gentoo.org</email> + <name>Tupone Alfredo</name> + </maintainer> + <maintainer type="person"> + <email>lockalsash@gmail.com</email> + <name>Sv. Lockal</name> + </maintainer> + <maintainer type="project" proxied="proxy"> + <email>proxy-maint@gentoo.org</email> + <name>Proxy Maintainers</name> + </maintainer> + <upstream> + <bugs-to>https://github.com/oneapi-src/oneDNN/issues</bugs-to> + <remote-id type="github">oneapi-src/oneDNN</remote-id> + </upstream> + <use> + <flag name="mkl">Use sci-libs/mkl for sgemm routines</flag> + <flag name="cblas">Use virtual/cblas for sgemm routines</flag> + </use> +</pkgmetadata> diff --git a/dev-libs/oneDNN/oneDNN-3.3.3.ebuild b/dev-libs/oneDNN/oneDNN-3.3.3.ebuild new file mode 100644 index 000000000000..88b11ae05e38 --- /dev/null +++ b/dev-libs/oneDNN/oneDNN-3.3.3.ebuild @@ -0,0 +1,106 @@ +# Copyright 2024 Gentoo Authors +# Distributed under the terms of the GNU General Public License v2 + +EAPI=8 + +DOCS_BUILDER="doxygen" +DOCS_DIR="${WORKDIR}/${P}_build" + +# oneDNN has its own FindBLAS.cmake file to find MKL (in a non-standard way). +# Removing of CMake modules is disabled. +CMAKE_REMOVE_MODULES_LIST=( none ) + +# There is additional sphinx documentation but we are missing dependency doxyrest. +inherit cmake docs + +DESCRIPTION="oneAPI Deep Neural Network Library" +HOMEPAGE="https://github.com/oneapi-src/oneDNN" +SRC_URI="https://github.com/oneapi-src/oneDNN/archive/refs/tags/v${PV}.tar.gz -> ${P}.tar.gz" + +LICENSE="Apache-2.0" +SLOT="0" +KEYWORDS="~amd64" + +IUSE="test cpu_flags_x86_avx512f cpu_flags_x86_avx2 cpu_flags_x86_sse4_1 mkl cblas static-libs" + +RESTRICT="!test? ( test )" + +DEPEND=" + mkl? ( sci-libs/mkl ) + cblas? ( !mkl? ( virtual/cblas ) ) +" +RDEPEND="${DEPEND}" + +src_configure() { + local mycmakeargs=( + -DDNNL_LIBRARY_TYPE=$(usex static-libs STATIC SHARED) + -DDNNL_CPU_RUNTIME=OMP + -DDNNL_GPU_RUNTIME=NONE + -DDNNL_BUILD_EXAMPLES=OFF + -DDNNL_BUILD_TESTS="$(usex test)" + -DDNNL_ENABLE_CONCURRENT_EXEC=OFF + -DDNNL_ENABLE_JIT_PROFILING=ON + -DDNNL_ENABLE_ITT_TASKS=ON + -DDNNL_ENABLE_PRIMITIVE_CACHE=ON + -DDNNL_ENABLE_MAX_CPU_ISA=ON + -DDNNL_ENABLE_CPU_ISA_HINTS=ON + -DDNNL_ENABLE_WORKLOAD=TRAINING + -DDNNL_ENABLE_PRIMITIVE=ALL + -DDNNL_ENABLE_PRIMITIVE_GPU_ISA=ALL + -DDNNL_EXPERIMENTAL=OFF + -DDNNL_VERBOSE=ON + -DDNNL_DEV_MODE=OFF + -DDNNL_AARCH64_USE_ACL=OFF + -DDNNL_GPU_VENDOR=INTEL + -DDNNL_LIBRARY_NAME=dnnl + -DONEDNN_BUILD_GRAPH=ON + -DONEDNN_ENABLE_GRAPH_DUMP=OFF + -DONEDNN_EXPERIMENTAL_GRAPH_COMPILER_BACKEND=OFF + -Wno-dev + ) + + local isa="ALL" + + if use cpu_flags_x86_avx512f ; then + isa="AVX512" + elif use cpu_flags_x86_avx2; then + isa="AVX2" + elif use cpu_flags_x86_sse4_1; then + isa="SSE41" + else + ewarn "WARNING: oneDNN is being built with for all ISA." + ewarn "These may cause runtime issues CPUs that are not supported by oneDNN." + ewarn "" + ewarn "To configure oneDNN with ISA that is optimal for your CPU," + ewarn "set CPU_FLAGS_X86 in your make.conf, and re-emerge oneDNN." + ewarn "" + ewarn "See the list of supported CPUs at" + ewarn "https://github.com/oneapi-src/oneDNN?tab=readme-ov-file#system-requirements" + ewarn "For CPU_FLAGS_X86 documentation visit https://wiki.gentoo.org/wiki/CPU_FLAGS_*" + fi + + mycmakeargs+=( -DDNNL_ENABLE_PRIMITIVE_CPU_ISA="${isa}" -DONEDNN_ENABLE_GEMM_KERNELS_ISA="${isa}" ) + + if use mkl ; then + source /opt/intel/oneapi/mkl/latest/env/vars.sh + mycmakeargs+=( -DDNNL_BLAS_VENDOR=MKL ) + elif use cblas; then + mycmakeargs+=( -DDNNL_BLAS_VENDOR=ANY -DBLA_VENDOR=Generic -DBLAS_LIBRARIES=-lcblas ) + else + mycmakeargs+=( -DDNNL_BLAS_VENDOR=NONE ) + fi + + cmake_src_configure +} + +src_compile() { + cmake_src_compile + docs_compile +} + +src_install() { + cmake_src_install + + # Correct docdir + mv "${ED}/usr/share/doc/dnnl"* "${ED}/usr/share/doc/${PF}" || die +} |