modify

dmlc · May 28, 2015 · 8bbed35 · 8bbed35
1 parent 9520b90
commit 8bbed35
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -4,7 +4,14 @@ endif
 export MPICXX = mpicxx
 export LDFLAGS= -Llib -lrt
 export WARNFLAGS= -Wall -Wextra -Wno-unused-parameter -Wno-unknown-pragmas -pedantic 
-export CFLAGS = -O3 -msse2 -fPIC $(WARNFLAGS) 
+export CFLAGS = -O3 -msse2 $(WARNFLAGS) 
+
+ifndef WITH_FPIC
+	WITH_FPIC = 1
+endif
+ifeq ($(WITH_FPIC), 1)
+	CFLAGS += -fPIC	
+endif
 
 # build path
 BPATH=.

diff --git a/guide/basic.cc b/guide/basic.cc
@@ -5,11 +5,15 @@
  *
  * \author Tianqi Chen
  */
+#include <vector>
 #include <rabit.h>
 using namespace rabit;
-const int N = 3;
 int main(int argc, char *argv[]) {
-  int a[N];
+  int N = 3;
+  if (argc > 1) {
+    N = atoi(argv[1]);
+  }
+  std::vector<int> a(N);
   rabit::Init(argc, argv);
   for (int i = 0; i < N; ++i) {
     a[i] = rabit::GetRank() + i;