| 3 |
3 |
|
| 4 |
4 |
EAPI=8
|
| 5 |
5 |
|
| 6 |
|
DISTUTILS_USE_PEP517=setuptools
|
|
6 |
DISTUTILS_USE_PEP517=scikit-build-core
|
| 7 |
7 |
PYTHON_COMPAT=( python3_{12..14} )
|
| 8 |
8 |
DISTUTILS_SINGLE_IMPL=1
|
| 9 |
9 |
DISTUTILS_EXT=1
|
| ... | ... | |
| 89 |
89 |
)
|
| 90 |
90 |
fbgemm? ( >=sci-ml/FBGEMM-1.4 )
|
| 91 |
91 |
gloo? ( >=sci-ml/gloo-2025.06.04[cuda?,rocm?] )
|
| 92 |
|
kineto? ( ~sci-ml/kineto-0.4.0_p20260603 )
|
|
92 |
kineto? ( ~sci-ml/kineto-0.4.0_p20260805 )
|
| 93 |
93 |
mimalloc? ( dev-libs/mimalloc )
|
| 94 |
94 |
mkl? ( sci-libs/mkl )
|
| 95 |
95 |
mpi? ( virtual/mpi )
|
| ... | ... | |
| 144 |
144 |
dev-libs/psimd
|
| 145 |
145 |
sci-ml/FP16
|
| 146 |
146 |
$(python_gen_cond_dep '
|
| 147 |
|
<dev-python/pybind11-3.0.5[${PYTHON_USEDEP}]
|
|
147 |
dev-python/pybind11[${PYTHON_USEDEP}]
|
| 148 |
148 |
dev-python/pyyaml[${PYTHON_USEDEP}]
|
| 149 |
149 |
dev-python/typing-extensions[${PYTHON_USEDEP}]
|
| 150 |
150 |
')
|
| 151 |
|
cuda? ( ~dev-libs/cutlass-4.4.2[tools(+)] )
|
|
151 |
cuda? ( ~dev-libs/cutlass-4.6.1[tools(+)] )
|
| 152 |
152 |
onednn? ( sci-ml/ideep )
|
| 153 |
153 |
rocm? (
|
| 154 |
154 |
>=sci-libs/hipCUB-6.3:= <sci-libs/hipCUB-7.3:=
|
| ... | ... | |
| 161 |
161 |
BDEPEND="dev-build/cmake"
|
| 162 |
162 |
|
| 163 |
163 |
PATCHES=(
|
| 164 |
|
"${FILESDIR}"/${P}-aotriton-fixes.patch
|
| 165 |
|
"${FILESDIR}"/${P}-glog.patch
|
| 166 |
|
"${FILESDIR}"/${P}-mimalloc.patch
|
|
164 |
"${FILESDIR}"/${PN}-2.13.0-aotriton-fixes.patch
|
|
165 |
"${FILESDIR}"/${PN}-2.13.0-glog.patch
|
|
166 |
"${FILESDIR}"/${PN}-2.13.0-mimalloc.patch
|
| 167 |
167 |
"${FILESDIR}"/${P}-removekineto-pr178960.patch
|
| 168 |
|
"${FILESDIR}"/${P}-rocm-assert-fix.patch
|
| 169 |
|
"${FILESDIR}"/${P}-unbundle_fbgemm.patch
|
| 170 |
|
"${FILESDIR}"/${P}-unbundle_flatbuffers.patch
|
| 171 |
|
"${FILESDIR}"/${P}-unbundle_fmt.patch
|
| 172 |
|
"${FILESDIR}"/${P}-unbundle_fxdiv.patch
|
| 173 |
|
"${FILESDIR}"/${P}-unbundle_httplib.patch
|
| 174 |
|
"${FILESDIR}"/${P}-unbundle_kineto.patch
|
| 175 |
|
"${FILESDIR}"/${P}-unbundle_mkldnn.patch
|
| 176 |
|
"${FILESDIR}"/${P}-unbundle_nnpack.patch
|
|
168 |
"${FILESDIR}"/${PN}-2.13.0-rocm-assert-fix.patch
|
|
169 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_fbgemm.patch
|
|
170 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_flatbuffers.patch
|
|
171 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_fmt.patch
|
|
172 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_fxdiv.patch
|
|
173 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_httplib.patch
|
|
174 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_kineto.patch
|
|
175 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_mkldnn.patch
|
|
176 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_nnpack.patch
|
| 177 |
177 |
"${FILESDIR}"/${P}-unbundle_pocketfft.patch
|
| 178 |
|
"${FILESDIR}"/${P}-xnnpack.patch
|
|
178 |
"${FILESDIR}"/${PN}-2.13.0-xnnpack.patch
|
|
179 |
"${FILESDIR}"/${P}-license.patch
|
|
180 |
"${FILESDIR}"/${P}-cpp-httplib.patch
|
| 179 |
181 |
)
|
| 180 |
182 |
|
| 181 |
183 |
src_prepare() {
|
| ... | ... | |
| 334 |
336 |
addpredict "/dev/char/"
|
| 335 |
337 |
|
| 336 |
338 |
local -x CMAKE_CUDA_FLAGS="$(cuda_gccdir -f | tr -d \")"
|
| 337 |
|
local -x TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-7.0}"
|
|
339 |
local -x TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-7.5}"
|
| 338 |
340 |
local -x USE_CUDNN=ON
|
| 339 |
341 |
local -x USE_FLASH_ATTENTION=OFF
|
| 340 |
342 |
local -x USE_MEM_EFF_ATTENTION=OFF
|
