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

%NAME%
python-pytorch-cuda

%BASE%
python-pytorch

%VERSION%
2.11.0-2

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

%CSIZE%
713607279

%ISIZE%
1258999817

%SHA256SUM%
355cfe1c02d408cd2e9ecbd355b9b4e0c9c1da52362e73e08b929c7cacee4969

%URL%
https://pytorch.org

%LICENSE%
BSD-3-Clause-Modification

%ARCH%
aarch64

%BUILDDATE%
1791449621

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

%CONFLICTS%
python-pytorch

%PROVIDES%
python-pytorch=2.11.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
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

