pkgbase = python-pytorch-rocm
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
	pkgver = 1.7.0rc2
	pkgrel = 1
	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.7.0rc2::git+https://github.com/pytorch/pytorch.git#tag=v1.7.0-rc2
	source = fix_include_system.patch
	source = find-hsa-runtime.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/45550.patch
	sha256sums = SKIP
	sha256sums = 147bdaeac8ec46ea46382e6146878bd8f8d51e05d5bd6f930dfd8e2b520859b9
	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