Skip to content

Commit

Permalink
Merge pull request #4326 from hqucms/mxnet/CMSSW_10_2_X/gcc700
Browse files Browse the repository at this point in the history
Backport MXNet and DeepBoostedJet models.
  • Loading branch information
cmsbuild authored Sep 27, 2018
2 parents e7580d3 + 95f4605 commit 97e2d99
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 11 deletions.
2 changes: 1 addition & 1 deletion data-RecoBTag-Combined.spec
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
### RPM cms data-RecoBTag-Combined V01-00-13
### RPM cms data-RecoBTag-Combined V01-00-14

%prep

Expand Down
2 changes: 1 addition & 1 deletion mxnet-predict-toolfile.spec
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
### RPM external mxnet-predict-toolfile 1.2.0
### RPM external mxnet-predict-toolfile 1.2.1.mod3
Requires: mxnet-predict
%prep

Expand Down
21 changes: 12 additions & 9 deletions mxnet-predict.spec
Original file line number Diff line number Diff line change
@@ -1,22 +1,25 @@
### RPM external mxnet-predict 1.2.0
%define projectname mxnet-predict
%define releasename apache-mxnet-src-%{realversion}-incubating
Source: https://github.com/apache/incubator-mxnet/releases/download/%{realversion}/apache-mxnet-src-%{realversion}-incubating.tar.gz
### RPM external mxnet-predict 1.2.1
%define tag 97171b96b2b7efc78eccfbe0a0c2561a377ce153
%define branch 1.2.1.mod3
%define github_user cms-externals
Source: git+https://github.com/%{github_user}/incubator-mxnet.git?obj=%{branch}/%{tag}&export=%{n}-%{realversion}&output=/%{n}-%{realversion}-%{tag}.tgz

Requires: OpenBLAS

%prep
%setup -q -n %releasename
%setup -q -n %{n}-%{realversion}

%build
make %{makeprocesses} USE_OPENCV=0 USE_OPENMP=0 USE_BLAS=openblas USE_CPP_PACKAGE=1 USE_F16C=0 \
ADD_LDFLAGS="-L$OPENBLAS_ROOT/lib" \
ADD_CFLAGS="-I$OPENBLAS_ROOT/include -DDISABLE_OPENMP=1 -DMSHADOW_RABIT_PS=0 -DMSHADOW_DIST_PS=0 -DMSHADOW_FORCE_STREAM -DMXNET_PREDICT_ONLY=1"
ADD_CFLAGS="-I$OPENBLAS_ROOT/include -DDISABLE_OPENMP=1 -DMSHADOW_RABIT_PS=0 -DMSHADOW_DIST_PS=0 -DMXNET_PREDICT_ONLY=1 -DMXNET_THREAD_LOCAL_ENGINE=1"

%install
mkdir -p %{i}/{lib,include}
mv lib/libmxnet.so %{i}/lib/libmxnetpredict.so
mv include/* %{i}/include
mv cpp-package/include/* %{i}/include
mv lib/libmxnet.so %{i}/lib/libmxnetpredict.so
mv include/* %{i}/include
mv 3rdparty/dmlc-core/include/* %{i}/include
mv 3rdparty/tvm/nnvm/include/* %{i}/include
mv cpp-package/include/* %{i}/include


0 comments on commit 97e2d99

Please sign in to comment.