%FILENAME%
python-pytorch-opt-rocm-2.3.1-4-x86_64.pkg.tar.zst

%NAME%
python-pytorch-opt-rocm

%BASE%
python-pytorch

%VERSION%
2.3.1-4

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

%CSIZE%
146878091

%ISIZE%
2446984829

%SHA256SUM%
2a781aec3a3a20f0945e8ca02f1ed9303c38da0ef0f9e30bdfe7d230cfb8c4b6

%URL%
https://pytorch.org

%LICENSE%
BSD

%ARCH%
x86_64

%BUILDDATE%
1719083282

%PACKAGER%
Antonio Rojas <arojas@archlinux.org>

%CONFLICTS%
python-pytorch

%PROVIDES%
python-pytorch=2.3.1
python-pytorch-rocm=2.3.1

%DEPENDS%
google-glog
gflags
opencv
openmp
openmpi
pybind11
python
python-yaml
libuv
python-numpy
python-sympy
protobuf
ffmpeg
python-future
qt6-base
eigen
intel-oneapi-mkl
python-typing_extensions
numactl
python-jinja
python-networkx
python-filelock
rocm-hip-sdk
hipblaslt
roctracer
miopen
magma-hip
onednn

%MAKEDEPENDS%
python
python-setuptools
python-yaml
python-numpy
cmake
cuda
gcc13
nccl
cudnn
git
rocm-hip-sdk
hipblaslt
roctracer
miopen
magma-cuda
magma-hip
ninja
pkgconfig
doxygen
vulkan-headers
shaderc
onednn

