blob: 7ae899dcbb27ae09440f51082228334ee89b9f2c (
plain)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
# Copyright 1999-2024 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
inherit cmake toolchain-funcs
DESCRIPTION="Common bricks library for building distributed machine learning"
HOMEPAGE="https://github.com/dmlc/dmlc-core"
if [[ ${PV} == *9999* ]] ; then
EGIT_REPO_URI="https://github.com/dmlc/${PN}.git"
inherit git-r3
else
SRC_URI="https://github.com/dmlc/${PN}/archive/refs/tags/v${PV}.tar.gz
-> ${P}.tar.gz"
KEYWORDS="~amd64 ~x86 ~amd64-linux ~x86-linux"
fi
LICENSE="Apache-2.0"
SLOT="0"
# hdfs needs big java hdfs not yet in portage
# azure not yet in portage
IUSE="cpu_flags_x86_sse2 doc openmp s3 test"
RESTRICT="!test? ( test )"
RDEPEND="s3? ( net-misc/curl[ssl] )"
DEPEND="${RDEPEND}"
BDEPEND="doc? ( app-text/doxygen[dot] )
test? ( dev-cpp/gtest )"
pkg_pretend() {
[[ ${MERGE_TYPE} != binary ]] && use openmp && tc-check-openmp
}
pkg_setup() {
[[ ${MERGE_TYPE} != binary ]] && use openmp && tc-check-openmp
}
src_prepare() {
cmake_src_prepare
sed -e '/-O3/d' -e '/check_cxx_compiler_flag("-msse2"/d' \
-e '/check_cxx.*SSE2/d' \
-i CMakeLists.txt || die
# All these hacks below to allow testing
sed -e 's|-O3||' -e 's|-std=c++11|-std=c++14|' \
-e "s|-lm|-lm -L\"${BUILD_DIR}\" -ldmlc|g" \
-i Makefile || die
sed -e "s|libdmlc.a||g" \
-e "/^GTEST_LIB=/s|=.*|=/usr/$(get_libdir)|" \
-e "/^GTEST_INC=/s|=.*|=/usr/include|" \
-i test/dmlc_test.mk test/unittest/dmlc_unittest.mk || die
# Don't ever download gtest
sed -e 's/^if (NOT GTEST_FOUND)$/if (FALSE)/' \
-i test/unittest/CMakeLists.txt || die
cat <<-EOF > config.mk
USE_SSE=$(usex cpu_flags_x86_sse2 1 0)
WITH_FPIC=1
USE_OPENMP=$(usex openmp 1 0)
USE_S3=$(usex s3 1 0)
BUILD_TEST=$(usex test 1 0)
DMLC_CFLAGS=${CXXFLAGS}
DMLC_LDFLAGS=${LDFLAGS}
EOF
}
src_configure() {
local mycmakeargs=(
-DGOOGLE_TEST=$(usex test)
-DSUPPORT_MSSE2=$(usex cpu_flags_x86_sse2)
-DUSE_CXX14_IF_AVAILABLE=YES # Newer gtest needs C++14 or later
-DUSE_S3=$(usex s3)
-DUSE_OPENMP=$(usex openmp)
)
cmake_src_configure
}
src_compile() {
cmake_src_compile
use doc && emake doxygen
use test && emake test
}
src_test() {
DMLC_UNIT_TEST_LITTLE_ENDIAN=$([[ $(tc-endian) == little ]] && echo 1 || echo 0) \
LD_LIBRARY_PATH="${BUILD_DIR}" \
test/unittest/dmlc_unittest || die
cmake_src_test
}
src_install() {
cmake_src_install
if use doc; then
dodoc -r doc/doxygen/html
docompress -x /usr/share/doc/${PF}/html
fi
}
|
GitWeb