blob: 702ac0c04740342f0f4c51a98e1609a58fd402e0 (
plain) (
blame)
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
|
PORTNAME= deepspeed
DISTVERSION= 0.19.6
CATEGORIES= misc python # machine-learning
MASTER_SITES= PYPI
PKGNAMEPREFIX= ${PYTHON_PKGNAMEPREFIX}
MAINTAINER= yuri@FreeBSD.org
COMMENT= DeepSpeed library for ChatGPT-like model training
WWW= http://www.deepspeed.ai/ \
https://github.com/deepspeedai/DeepSpeed
LICENSE= APACHE20
BUILD_DEPENDS= bash:shells/bash
RUN_DEPENDS= ${PYTHON_PKGNAMEPREFIX}einops>0:misc/py-einops@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}hjson>0:textproc/py-hjson@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}msgpack>0:devel/py-msgpack@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}ninja>0:devel/py-ninja@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}numpy>0:math/py-numpy@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}packaging>=20.0:devel/py-packaging@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}psutil>0:sysutils/py-psutil@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}py-cpuinfo2>0:sysutils/py-py-cpuinfo2@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}pydantic2>=2.0.0:devel/py-pydantic2@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}pytorch>=2.0.0:misc/py-pytorch@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}tqdm>0:misc/py-tqdm@${PY_FLAVOR} \
bash:shells/bash
# many extras exist in setup.py
USES= python shebangfix
USE_PYTHON= distutils autoplist pytest # some tests fail, tests consume gigabytes of disk space
SHEBANG_FILES= bin/ds_ssh
TEST_ENV= ${MAKE_ENV} PYTHONPATH=${STAGEDIR}${PYTHONPREFIX_SITELIBDIR}
TEST_ARGS= --disable-plugin-autoload
NO_ARCH= yes
# tests as of 0.19.6: 2 failed, 44 passed, 1 warning in 2.87s, see https://github.com/deepspeedai/DeepSpeed/issues/8554
.include <bsd.port.mk>
|
