Skip to content

Commit

Permalink
patch cutorch to fix build on ARM
Browse files Browse the repository at this point in the history
  • Loading branch information
dusty-nv committed Sep 28, 2016
1 parent 47ca8ba commit 62af1a1
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions CMakePreBuild.sh
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,11 @@ $TORCH_PREFIX/bin/luarocks install $BUILD_ROOT/rocks/gnuplot-scm-1.rockspec
#$TORCH_PREFIX/bin/luarocks install gnuplot

git clone https://github.com/torch/cutorch
sed -i 's/-j$(getconf _NPROCESSORS_ONLN)/-j1/g' cutorch/rocks/cutorch-1.0-0.rockspec
sed -i 's/-j$(getconf _NPROCESSORS_ONLN)/-j1/g' cutorch/rocks/cutorch-scm-1.rockspec
sed -i 's/$(getconf _NPROCESSORS_ONLN)/1/g' cutorch/rocks/cutorch-1.0-0.rockspec
sed -i 's/$(getconf _NPROCESSORS_ONLN)/1/g' cutorch/rocks/cutorch-scm-1.rockspec
sed -i 's/jopts=3/jopts=1/g' cutorch/rocks/cutorch-1.0-0.rockspec
sed -i 's/jopts=3/jopts=1/g' cutorch/rocks/cutorch-scm-1.rockspec

$TORCH_PREFIX/bin/luarocks install $BUILD_ROOT/cutorch/rocks/cutorch-scm-1.rockspec

# install cudnn v5 bindings
Expand Down

0 comments on commit 62af1a1

Please sign in to comment.