%FILENAME%
python-pytorch-rocm-2.12.0-2-aarch64.pkg.tar.zst

%NAME%
python-pytorch-rocm

%BASE%
python-pytorch

%VERSION%
2.12.0-2

%DESC%
Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCm)

%CSIZE%
229510563

%ISIZE%
638807578

%SHA256SUM%
25c0b02be493f114ab809cd7aa5e22e600a8bda21cb951297425b443ea9e421e

%URL%
https://pytorch.org

%LICENSE%
BSD-3-Clause-Modification

%ARCH%
aarch64

%BUILDDATE%
1793577580

%PACKAGER%
GitLab CI Package Builder <ci-package-builder-1@steamos.cloud>

%CONFLICTS%
python-pytorch

%PROVIDES%
python-pytorch=2.12.0

%DEPENDS%
abseil-cpp
eigen
gflags
glibc
google-glog
libgcc
libgomp
libstdc++
libuv
numactl
openmp
openmpi
protobuf
pybind11
python
python-filelock
python-fsspec
python-jinja
python-networkx
python-numpy
python-sympy
python-typing_extensions
python-yaml
qt6-base
vulkan-icd-loader
amdsmi
rocm-hip-sdk
hipblaslt
rocprofiler
roctracer
miopen-hip
onednn
python-triton
python-aotriton

%OPTDEPENDS%
python-onnxscript: for the ONNX exporter

%MAKEDEPENDS%
amdsmi
cmake
cuda
cudnn
doxygen
git
hipblaslt
miopen-hip
nccl
ninja
onednn
pkgconfig
python-aotriton
python-build
python-installer
python-setuptools
python-triton
python-yaml
rocm-hip-sdk
rocm-toolchain
roctracer
procps-ng
shaderc
vulkan-headers

