pkgbase = python-pytorch-git
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
	pkgver = 1.3.1.r22820.1350b99de4
	pkgrel = 1
	url = https://pytorch.org
	arch = x86_64
	license = BSD
	makedepends = cmake
	makedepends = cuda
	makedepends = cuda
	makedepends = cudnn
	makedepends = cudnn
	makedepends = doxygen
	makedepends = git
	makedepends = magma
	makedepends = nccl
	makedepends = python-numpy
	makedepends = python-setuptools
	makedepends = python-yaml
	depends = ffmpeg
	depends = gflags
	depends = google-glog
	depends = leveldb
	depends = lmdb
	depends = opencv
	depends = openmp
	depends = protobuf
	depends = pybind11
	depends = python-future
	depends = python-numpy
	depends = python-yaml
	depends = qt5-base
	depends = zeromq
	source = pytorch::git+https://github.com/pytorch/pytorch.git
	sha512sums = SKIP

pkgname = python-pytorch-git

pkgname = python-pytorch-mkl-git
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with MKL-DNN)
	provides = python-pytorch=1.3.1.r22820.1350b99de4
	conflicts = python-pytorch

pkgname = python-pytorch-cuda-git
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CUDA)
	depends = ffmpeg
	depends = gflags
	depends = google-glog
	depends = leveldb
	depends = lmdb
	depends = opencv
	depends = openmp
	depends = protobuf
	depends = pybind11
	depends = python-future
	depends = python-numpy
	depends = python-yaml
	depends = qt5-base
	depends = zeromq
	depends = cuda
	depends = cudnn
	depends = magma
	depends = nccl
	provides = python-pytorch=1.3.1.r22820.1350b99de4
	conflicts = python-pytorch

pkgname = python-pytorch-mkl-cuda-git
	pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CUDA and MKL-DNN)
	depends = ffmpeg
	depends = gflags
	depends = google-glog
	depends = leveldb
	depends = lmdb
	depends = opencv
	depends = openmp
	depends = protobuf
	depends = pybind11
	depends = python-future
	depends = python-numpy
	depends = python-yaml
	depends = qt5-base
	depends = zeromq
	depends = cuda
	depends = cudnn
	depends = magma
	depends = nccl
	provides = python-pytorch=1.3.1.r22820.1350b99de4
	provides = python-pytorch-cuda=1.3.1.r22820.1350b99de4
	conflicts = python-pytorch