You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
2.2 KiB
98 lines
2.2 KiB
# Copyright 1999-2021 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=6
|
|
|
|
inherit toolchain-funcs cmake-utils
|
|
|
|
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
|
|
EGIT_COMMIT="54db57d5d1b2a7b93319053011802888b827a539"
|
|
inherit vcs-snapshot
|
|
KEYWORDS="~amd64 ~x86 ~amd64-linux ~x86-linux"
|
|
SRC_URI="https://github.com/dmlc/dmlc-core/archive/${EGIT_COMMIT}.tar.gz -> ${P}.tar.gz"
|
|
fi
|
|
|
|
LICENSE="Apache-2.0"
|
|
SLOT="0"
|
|
|
|
# hdfs needs big java hdfs not yet in portage
|
|
# azure not yet in portage
|
|
IUSE="doc openmp s3 test"
|
|
RESTRICT="!test? ( test )"
|
|
|
|
RDEPEND="net-misc/curl[ssl]"
|
|
DEPEND="${RDEPEND}
|
|
doc? ( app-doc/doxygen )
|
|
test? ( dev-cpp/gtest )
|
|
"
|
|
|
|
PATCHES=( "${FILESDIR}"/${PN}-install-dirs.patch )
|
|
|
|
pkg_pretend() {
|
|
[[ ${MERGE_TYPE} != binary ]] && use openmp && tc-check-openmp
|
|
}
|
|
|
|
pkg_setup() {
|
|
[[ ${MERGE_TYPE} != binary ]] && use openmp && tc-check-openmp
|
|
}
|
|
|
|
src_prepare() {
|
|
cmake-utils_src_prepare
|
|
# respect user flags (SSE2 does nothing more than adding -msse2)
|
|
# also doc install everything so remove
|
|
sed -e '/-O3/d' \
|
|
-e '/check_cxx.*SSE2/d' \
|
|
-i CMakeLists.txt || die
|
|
# all these hacks below to allow testing
|
|
sed -e 's|-O3||' -e 's|-lm|-lm -L$(LD_LIBRARY_PATH) -ldmlc|g' -i Makefile || die
|
|
sed -e "s|libdmlc.a||g" \
|
|
-i test/dmlc_test.mk test/unittest/dmlc_unittest.mk || die
|
|
cat <<-EOF > config.mk
|
|
USE_SSE=0
|
|
WITH_FPIC=1
|
|
USE_OPENMP=$(use openmp && echo 1 || echo 0)
|
|
USE_S3=$(use s3 && echo 1 || echo 0)
|
|
BUILD_TEST=$(use test && echo 1 || echo 0)
|
|
DMLC_CFLAGS=${CXXFLAGS}
|
|
DMLC_LDFLAGS=${LDFLAGS}
|
|
EOF
|
|
}
|
|
|
|
src_configure() {
|
|
local mycmakeargs=(
|
|
-DBUILD_SHARED_LIBS=ON
|
|
-DUSE_S3="$(usex s3)"
|
|
-DUSE_OPENMP="$(usex openmp)"
|
|
)
|
|
cmake-utils_src_configure
|
|
}
|
|
|
|
src_compile() {
|
|
cmake-utils_src_compile
|
|
if use doc; then
|
|
doxygen doc/Doxyfile || die
|
|
fi
|
|
}
|
|
|
|
src_test() {
|
|
tc-export CXX
|
|
export LD_LIBRARY_PATH="${BUILD_DIR}"
|
|
emake test
|
|
test/unittest/dmlc_unittest || die
|
|
}
|
|
|
|
src_install() {
|
|
cmake-utils_src_install
|
|
if use doc; then
|
|
insinto /usr/share/doc/${PF}
|
|
doins -r doc/doxygen/html
|
|
docompress -x /usr/share/doc/${PF}/html
|
|
fi
|
|
}
|