зеркало из https://github.com/microsoft/caffe.git
Merge pull request #2224 from small-yellow-duck/master
[build] check if CPU_ONLY is set when determining CUDA version
This commit is contained in:
Коммит
2a7fe03b5d
14
Makefile
14
Makefile
|
@ -232,13 +232,15 @@ endif
|
|||
# libstdc++ for NVCC compatibility on OS X >= 10.9 with CUDA < 7.0
|
||||
ifeq ($(OSX), 1)
|
||||
CXX := /usr/bin/clang++
|
||||
CUDA_VERSION := $(shell $(CUDA_DIR)/bin/nvcc -V | grep -o 'release \d' | grep -o '\d')
|
||||
ifeq ($(shell echo $(CUDA_VERSION) \< 7.0 | bc), 1)
|
||||
CXXFLAGS += -stdlib=libstdc++
|
||||
LINKFLAGS += -stdlib=libstdc++
|
||||
ifneq ($(CPU_ONLY), 1)
|
||||
CUDA_VERSION := $(shell $(CUDA_DIR)/bin/nvcc -V | grep -o 'release \d' | grep -o '\d')
|
||||
ifeq ($(shell echo $(CUDA_VERSION) \< 7.0 | bc), 1)
|
||||
CXXFLAGS += -stdlib=libstdc++
|
||||
LINKFLAGS += -stdlib=libstdc++
|
||||
endif
|
||||
# clang throws this warning for cuda headers
|
||||
WARNINGS += -Wno-unneeded-internal-declaration
|
||||
endif
|
||||
# clang throws this warning for cuda headers
|
||||
WARNINGS += -Wno-unneeded-internal-declaration
|
||||
# gtest needs to use its own tuple to not conflict with clang
|
||||
COMMON_FLAGS += -DGTEST_USE_OWN_TR1_TUPLE=1
|
||||
# boost::thread is called boost_thread-mt to mark multithreading on OS X
|
||||
|
|
Загрузка…
Ссылка в новой задаче