File pytorch.spec of Package pytorch

%define	prefix	/usr

Name:          pytorch
Version:       1.6.0
Release:       0
Summary:       Collection of algorithms for deep learning
Group:         Development/Libraries
URL:           https://github.com/pytorch
Source:        %{name}-%{version}.tar.gz
License:       BSD 
BuildRequires: gflags-devel glog-devel protobuf-devel protobuf-compiler
BuildRequires: cmake python3-devel python3-pip python3-yaml
BuildRequires: dos2unix
BuildRoot:     %{_tmppath}/%{name}-%{version}-root
#Patch0: 21737.patch
Prefix: %prefix

%description
PyTorch - lib for deep learning

%prep
%setup -q
# find . -type f -print0 | xargs -0 dos2unix --
# %patch0 -p0
%build
#pip3 install future setuptools typing_extensions future six requests pyyaml

CAFFE2_ROOT=`pwd`
BUILD_ROOT=${BUILD_ROOT:-"$CAFFE2_ROOT/build_mobile"}
INSTALL_PREFIX=${BUILD_ROOT}/install
mkdir -p $INSTALL_PREFIX

cd ${BUILD_ROOT}

cmake -DCMAKE_INSTALL_PREFIX=%{prefix} \
      -DCMAKE_BUILD_TYPE=Release       \
      -DUSE_STATIC_DISPATCH=ON  \
 -DBUILD_MOBILE_BENCHMARK=1 \
-DBUILD_MOBILE_TEST=1 \
-DBUILD_TEST=OFF \
-DBUILD_BINARY=OFF \
-DBUILD_PYTHON=OFF \
-DBUILD_SHARED_LIBS=OFF \
 -DUSE_CUDA=OFF \
 -DUSE_GFLAGS=OFF \
 -DUSE_OPENCV=OFF \
 -DUSE_LMDB=OFF \
 -DUSE_LEVELDB=OFF \
 -DUSE_MPI=OFF \
 -DUSE_OPENMP=OFF \
 -DUSE_DISTRIBUTED=OFF \
 -DUSE_VULKAN=OFF \
      -DCMAKE_PREFIX_PATH=$(python3 -c 'from distutils.sysconfig import get_python_lib; print(get_python_lib())') \
      -DPYTHON_EXECUTABLE=$(python3 -c 'import sys; print(sys.executable)') \
      -Wno-dev  $CAFFE2_ROOT
      
make

%install
CAFFE2_ROOT=`pwd`
echo "Caffe2 path: $CAFFE2_ROOT"
BUILD_ROOT=${BUILD_ROOT:-"$CAFFE2_ROOT/build_mobile"}

cd ${BUILD_ROOT}
rm -rf $RPM_BUILD_ROOT/*

make DESTDIR=%{buildroot} install
ls ${BUILD_ROOT}
ls ${BUILD_ROOT}/lib
#Some weird issue. Doesn't install all libs.
cp ${BUILD_ROOT}/lib/*.a %{buildroot}%{_libdir}

%clean
rm -rf "%{buildroot}"

%post -p /sbin/ldconfig
%postun -p /sbin/ldconfig

%files
%defattr(-,root,root)

%dir %{_includedir}/google
%dir %{_includedir}/nomnigraph
%dir %{_includedir}/pybind11
%{_includedir}/google/*
%{_includedir}/nomnigraph/*
%{_includedir}/pybind11/*
%{_libdir}/*.so
%dir %{_libdir}/cmake/protobuf
%{_libdir}/pkgconfig/protobuf-lite.pc

%{_libdir}/pkgconfig/protobuf.pc
%{_libdir}/cmake/protobuf/*
%{_bindir}/protoc-3.11.4.0
%{_bindir}/protoc
%dir %{_includedir}/torch
%dir %{_includedir}/THCUNN
%dir %{_includedir}/TH
%dir %{_includedir}/fp16
%dir %{_includedir}/caffe2
%dir %{_includedir}/c10
%dir %{_includedir}/ATen
%{_includedir}/torch/*
%{_includedir}/THCUNN/*
%{_includedir}/TH/*
%{_includedir}/fp16/*
%{_includedir}/caffe2/*
%{_includedir}/c10/*
%{_includedir}/ATen/*
%{_includedir}/*.h
%{_libdir}/*.a
%{_datadir}/cmake
%{_datadir}/ATen

%changelog