%FILENAME%
python-pytorch-cuda-2.9.1-2-aarch64.pkg.tar.zst

%NAME%
python-pytorch-cuda

%BASE%
python-pytorch

%VERSION%
2.9.1-2

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

%CSIZE%
693722688

%ISIZE%
1169305611

%SHA256SUM%
ce6ad270fab3b3a47f28fdf8612709ad5b1cc92a7e606346c28a87b79907e237

%URL%
https://pytorch.org

%LICENSE%
BSD-3-Clause-Modification

%ARCH%
aarch64

%BUILDDATE%
1783372404

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

%CONFLICTS%
python-pytorch

%PROVIDES%
python-pytorch=2.9.1

%DEPENDS%
abseil-cpp
eigen
gcc-libs
gflags
glibc
google-glog
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
cuda
nccl
cudnn
magma-cuda
onednn

%OPTDEPENDS%
python-onnxscript: for the ONNX exporter

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

