hide sse2 behing USE_SSE (apache#164)

stefanhenneking · Sep 14, 2016 · 64b6ab0 · 64b6ab0
1 parent ca4d4f6
commit 64b6ab0
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/make/mshadow.mk b/make/mshadow.mk
@@ -8,10 +8,21 @@
 #  Add MSHADOW_NVCCFLAGS to the nvcc compile flags
 #----------------------------------------------------------------------------------------
 
-MSHADOW_CFLAGS = -msse3 -funroll-loops -Wno-unused-parameter -Wno-unknown-pragmas
+MSHADOW_CFLAGS = -funroll-loops -Wno-unused-parameter -Wno-unknown-pragmas
 MSHADOW_LDFLAGS = -lm
 MSHADOW_NVCCFLAGS =
 MKLROOT =
+
+ifndef USE_SSE
+	USE_SSE=1
+endif
+
+ifeq ($USE_SSE), 1)
+	MSHADOW_CFLAGS += -msse3
+else
+	MSHADOW_CFLAGS += -DMSHADOW_USE_SSE=0
+endif
+
 ifeq ($(USE_CUDA), 0)
 	MSHADOW_CFLAGS += -DMSHADOW_USE_CUDA=0
 else