SINGA-104 Add Context Class
Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/35de4f91 Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/35de4f91 Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/35de4f91 Branch: refs/heads/master Commit: 35de4f913e97c64dbf27ff37c999aaa5a3ce40f6 Parents: 771ff32 Author: seaok <[email protected]> Authored: Thu Nov 26 10:51:55 2015 +0800 Committer: seaok <[email protected]> Committed: Thu Nov 26 10:51:55 2015 +0800 ---------------------------------------------------------------------- Makefile.gpu | 2 ++ 1 file changed, 2 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/35de4f91/Makefile.gpu ---------------------------------------------------------------------- diff --git a/Makefile.gpu b/Makefile.gpu index 2dc2a71..2fea3b2 100644 --- a/Makefile.gpu +++ b/Makefile.gpu @@ -57,6 +57,8 @@ CUCXXFLAGS := $(MSHADOW_FLAGS) -std=c++11 -G $(CUDA_ARCH) \ ifeq ($(CUDA_DIR),) MSHADOW_FLAGS := $(MSHADOW_FLAGS) -DCPU_ONLY CXXFLAGS := $(CXXFLAGS) -DCPU_ONLY +else + CXXFLAGS := $(CXXFLAGS) -DUSE_GPU endif # find user defined .proto file, and then compute the corresponding .h, .cc
