travis (wip): set -j4 on travis to prevent from out-of-memory

dmlc · Sep 6, 2017 · 0d274db · 0d274db
1 parent 35d1ec2
commit 0d274db
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/deps/build.jl b/deps/build.jl
@@ -54,6 +54,14 @@ else
   info("Did not find a CUDA installation, using CPU-only version of MXNet.")
 end
 
+function get_cpucore()
+    if haskey(ENV, "TRAVIS")  # on travis-ci
+        4
+    else
+        min(Sys.CPU_CORES, 8)
+    end
+end
+
 using BinDeps
 @BinDeps.setup
 if !libmxnet_detected
@@ -170,9 +178,9 @@ if !libmxnet_detected
           ChangeDirectory(_mxdir)
           `cp ../../cblas.h include/cblas.h`
           if USE_JULIA_BLAS
-            `make -j$(min(Sys.CPU_CORES,8)) USE_BLAS=$blas_name $MSHADOW_LDFLAGS`
+            `make -j$(get_cpucore()) USE_BLAS=$blas_name $MSHADOW_LDFLAGS`
           else
-            `make -j$(min(Sys.CPU_CORES,8))`
+            `make -j$(get_cpucore())`
           end
         end
         FileRule(joinpath(_libdir, "libmxnet.so"), @build_steps begin