blob: dfebd8855d15d9d12190c005de5314575eb67850 (
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
106
107
108
109
110
111
112
113
114
|
# Copyright 2024-2025 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
PYTHON_COMPAT=( python3_{10..13} )
DISTUTILS_SINGLE_IMPL=1
DISTUTILS_USE_PEP517=setuptools
DISTUTILS_EXT=1
ROCM_SKIP_GLOBALS=1
inherit cuda distutils-r1 multiprocessing rocm
DESCRIPTION="Datasets, transforms and models to specific to computer vision"
HOMEPAGE="https://github.com/pytorch/vision"
SRC_URI="https://github.com/pytorch/vision/archive/refs/tags/v${PV}.tar.gz
-> ${P}.tar.gz"
S="${WORKDIR}"/vision-${PV}
LICENSE="BSD"
SLOT="0"
KEYWORDS="~amd64"
IUSE="cuda +ffmpeg +jpeg +png rocm +webp"
REQUIRED_USE="
?? ( cuda rocm )
"
RDEPEND="
$(python_gen_cond_dep '
dev-python/numpy[${PYTHON_USEDEP}]
dev-python/pillow[${PYTHON_USEDEP}]
')
jpeg? ( media-libs/libjpeg-turbo:= )
png? ( media-libs/libpng:= )
webp? ( media-libs/libwebp )
ffmpeg? ( media-video/ffmpeg )
sci-ml/caffe2[cuda?,rocm?,${PYTHON_SINGLE_USEDEP}]
sci-ml/pytorch[${PYTHON_SINGLE_USEDEP}]
"
BDEPEND="
test? (
$(python_gen_cond_dep '
dev-python/pytest-mock[${PYTHON_USEDEP}]
dev-python/lmdb[${PYTHON_USEDEP}]
')
)
"
distutils_enable_tests pytest
PATCHES=( "${FILESDIR}"/${PN}-0.24.0-ffmpeg8.patch )
src_prepare() {
# multilib fixes
sed "s/ffmpeg_root, \"lib\"/ffmpeg_root, \"$(get_libdir)\"/" \
-i setup.py || die
use cuda && cuda_src_prepare
distutils-r1_src_prepare
}
src_configure() {
rocm_add_sandbox -w
distutils-r1_src_configure
}
python_compile() {
addpredict /dev/kfd
export FORCE_CUDA=0
if use cuda || use rocm ; then
export FORCE_CUDA=1
fi
export TORCHVISION_USE_PNG=$(usex png 1 0)
export TORCHVISION_USE_JPEG=$(usex jpeg 1 0)
export TORCHVISION_USE_WEBP=$(usex webp 1 0)
export TORCHVISION_USE_FFMPEG=$(usex ffmpeg 1 0)
export TORCHVISION_USE_NVJPEG=$(usex cuda 1 0)
export TORCHVISION_USE_VIDEO_CODEC=$(usex cuda 1 0)
NVCC_FLAGS="${NVCCFLAGS}" \
MAX_JOBS="$(makeopts_jobs)" \
distutils-r1_python_compile -j1
}
python_test() {
rm -r torchvision || die
local EPYTEST_IGNORE=(
test/test_videoapi.py
)
local EPYTEST_DESELECT=(
test/test_backbone_utils.py::TestFxFeatureExtraction::test_forward_backward
test/test_backbone_utils.py::TestFxFeatureExtraction::test_jit_forward_backward
test/test_models.py::test_classification_model
test/test_extended_models.py::TestHandleLegacyInterface::test_pretrained_pos
test/test_extended_models.py::TestHandleLegacyInterface::test_equivalent_behavior_weights
test/test_image.py::test_decode_avif[decode_avif]
test/test_image.py::test_decode_bad_encoded_data
test/test_image.py::test_decode_gif[False-earth]
test/test_image.py::test_decode_gif[True-earth]
test/test_image.py::test_decode_heic[decode_heic]
test/test_image.py::test_decode_webp
test/test_models.py::test_quantized_classification_model
test/test_ops.py::test_roi_opcheck
test/test_ops.py::TestDeformConv::test_aot_dispatch_dynamic__test_backward
test/test_ops.py::TestDeformConv::test_aot_dispatch_dynamic__test_forward
)
epytest
}
|
