pkgbase = python-pytorch-rocm
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
	pkgver = 1.6.0
	pkgrel = 3
	url = https://pytorch.org
	arch = x86_64
	license = BSD
	makedepends = python
	makedepends = python-setuptools
	makedepends = python-yaml
	makedepends = python-numpy
	makedepends = cmake
	makedepends = rocm
	makedepends = rocm-libs
	makedepends = miopen
	makedepends = git
	makedepends = magma
	makedepends = ninja
	makedepends = pkgconfig
	makedepends = doxygen
	depends = google-glog
	depends = gflags
	depends = opencv
	depends = openmp
	depends = rccl
	depends = pybind11
	depends = python
	depends = python-yaml
	depends = libuv
	depends = python-numpy
	depends = protobuf
	depends = ffmpeg
	depends = python-future
	depends = qt5-base
	depends = onednn
	depends = intel-mkl
	source = pytorch-1.6.0::git+https://github.com/pytorch/pytorch.git#tag=v1.6.0
	source = fix_include_system.patch
	source = use-system-libuv.patch
	source = use-system-libuv2.patch
	source = nccl_version.patch
	source = find-hip.patch
	source = find-rccl.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/42072.patch
	sha256sums = SKIP
	sha256sums = 147bdaeac8ec46ea46382e6146878bd8f8d51e05d5bd6f930dfd8e2b520859b9
	sha256sums = 6f3b7a87172011de810bf1ab581245b4463ef86e5cd09bec63aeffa372e26646
	sha256sums = 7b65c3b209fc39f92ba58a58be6d3da40799f1922910b1171ccd9209eda1f9eb
	sha256sums = 1a276bd827a0c76dab908cbc6605fa4c9fc2cc2b9431b6578a41133ae27dba2b
	sha256sums = c41b5a95513d8ff533eb083eee8e9106553669ef6fe4da238729d5dbd92f969f
	sha256sums = SKIP

pkgname = python-pytorch-rocm
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM)
	depends = google-glog
	depends = gflags
	depends = opencv
	depends = openmp
	depends = rccl
	depends = pybind11
	depends = python
	depends = python-yaml
	depends = libuv
	depends = python-numpy
	depends = protobuf
	depends = ffmpeg
	depends = python-future
	depends = qt5-base
	depends = onednn
	depends = intel-mkl
	depends = rocm
	depends = magma
	provides = python-pytorch
	conflicts = python-pytorch

pkgname = python-pytorch-opt-rocm
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM and CPU optimizations)
	depends = google-glog
	depends = gflags
	depends = opencv
	depends = openmp
	depends = rccl
	depends = pybind11
	depends = python
	depends = python-yaml
	depends = libuv
	depends = python-numpy
	depends = protobuf
	depends = ffmpeg
	depends = python-future
	depends = qt5-base
	depends = onednn
	depends = intel-mkl
	depends = rocm
	depends = magma
	provides = python-pytorch
	provides = python-pytorch-rocm
	conflicts = python-pytorch