| 29 |
29 |
SRC_URI="https://github.com/pytorch/${PN}/archive/refs/tags/v${PV}.tar.gz
|
| 30 |
30 |
-> ${P}.tar.gz
|
| 31 |
31 |
rocm? (
|
| 32 |
|
https://github.com/AngryLoki/pytorch/commit/bcb99546b6a6b57742eb0ec1705e55981ed071be.patch
|
| 33 |
|
-> ${PN}-2.14.1-aotriton-0.14.patch
|
| 34 |
32 |
https://github.com/ROCm/composable_kernel/archive/${CK_COMMIT}.tar.gz
|
| 35 |
33 |
-> ${CK_P}.tar.gz
|
| 36 |
34 |
https://github.com/ROCm/rocm-libraries/commit/0eab68951d3bb668e8f99beba2e2c07d7e1ca18a.patch
|
| ... | ... | |
| 167 |
165 |
PATCHES=(
|
| 168 |
166 |
"${FILESDIR}"/${PN}-2.13.0-aotriton-fixes.patch
|
| 169 |
167 |
"${FILESDIR}"/${PN}-2.13.0-glog.patch
|
| 170 |
|
"${FILESDIR}"/${PN}-2.13.0-mimalloc.patch
|
|
168 |
"${FILESDIR}"/${P}-mimalloc.patch
|
| 171 |
169 |
"${FILESDIR}"/${PN}-2.14.0-removekineto-pr178960.patch
|
| 172 |
170 |
"${FILESDIR}"/${PN}-2.13.0-rocm-assert-fix.patch
|
| 173 |
171 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_fbgemm.patch
|
| ... | ... | |
| 175 |
173 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_fmt.patch
|
| 176 |
174 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_fxdiv.patch
|
| 177 |
175 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_httplib.patch
|
| 178 |
|
"${FILESDIR}"/${PN}-2.13.0-unbundle_kineto.patch
|
|
176 |
"${FILESDIR}"/${P}-unbundle_kineto.patch
|
| 179 |
177 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_mkldnn.patch
|
| 180 |
178 |
"${FILESDIR}"/${PN}-2.13.0-unbundle_nnpack.patch
|
| 181 |
179 |
"${FILESDIR}"/${PN}-2.14.0-unbundle_pocketfft.patch
|
| 182 |
180 |
"${FILESDIR}"/${PN}-2.13.0-xnnpack.patch
|
| 183 |
|
"${FILESDIR}"/${PN}-2.14.0-license.patch
|
|
181 |
"${FILESDIR}"/${P}-license.patch
|
| 184 |
182 |
"${FILESDIR}"/${PN}-2.14.0-cpp-httplib.patch
|
| 185 |
183 |
)
|
| 186 |
184 |
|
| ... | ... | |
| 251 |
249 |
CMakeLists.txt
|
| 252 |
250 |
|
| 253 |
251 |
if use rocm; then
|
| 254 |
|
eapply "${DISTDIR}/${PN}-2.14.1-aotriton-0.14.patch"
|
| 255 |
|
|
| 256 |
252 |
# TODO: delete, when caffe2 depends on systemwide composable_kernel
|
| 257 |
253 |
sed -e "s:third_party/composable_kernel:../composable_kernel-${CK_COMMIT}:g" \
|
| 258 |
254 |
-i aten/src/ATen/CMakeLists.txt || die
|
| ... | ... | |
| 266 |
262 |
-i torch/headeronly/util/Half.h \
|
| 267 |
263 |
-i aten/src/ATen/native/cuda/int4mm.cu || die
|
| 268 |
264 |
|
| 269 |
|
# hipFile (only for torch.cuda.gds) is not packaged
|
|
265 |
# hipFile (only for torch.cuda.gds) is not packaged, amd_smi is needed only for distributed
|
| 270 |
266 |
sed -e 's/find_package_and_print_version(hipfile REQUIRED)/find_package_and_print_version(hipfile)/' \
|
|
267 |
-e 's/find_package_and_print_version(amd_smi REQUIRED)/find_package_and_print_version(amd_smi)/' \
|
| 271 |
268 |
-i cmake/public/LoadHIP.cmake || die
|
| 272 |
269 |
|
| 273 |
270 |
ebegin "HIPifying cuda sources"
|
