-
Notifications
You must be signed in to change notification settings - Fork 262
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
CUDNN v4 upgrade. Optionally, CNMEM used for GPU pool allocation. #42
Conversation
Add argmax_param "axis" to maximise output along the specified axis
488f126 Conflicts: Makefile Makefile.config.example src/caffe/layers/cudnn_conv_layer.cpp src/caffe/layers/cudnn_conv_layer.cu src/caffe/layers/cudnn_lcn_layer.cpp src/caffe/layers/cudnn_lcn_layer.cu
438add5 Conflicts: CMakeLists.txt cmake/Templates/CaffeConfig.cmake.in
38d6baf Conflicts: src/caffe/layers/cudnn_conv_layer.cpp src/caffe/layers/cudnn_conv_layer.cu
dummy alloc / free in MemoryHandlerActivator to ensure that the memory pool has been set up before any potential operations
add badge for travis build and license
Useful for exporting models from MATLAB (e.g. MatConvNet) to Caffe
Add a caffe.io.write_mean function to the MATLAB interface
SilenceLayer Backward bugfix (fixes BVLC#3151)
…fixes Documentation References ImageNet Sample Location
git-subtree-dir: .3rdparty/cnmem git-subtree-split: e817a7a2889389eaf934fe66a9744878b900acba
@@ -8,6 +8,9 @@ env: | |||
- WITH_CUDA=true WITH_CMAKE=true WITH_IO=true | |||
- WITH_CUDA=false WITH_CMAKE=false WITH_IO=false | |||
- WITH_CUDA=false WITH_CMAKE=true WITH_IO=false PYTHON_VERSION=3 | |||
# Currently there is no way to install cudnn via apt-get. Uncomment wjen it's available. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
check spelling
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed.
Woohoo, all the tests are passing! Now lets clean up the commit history and get this merged!
We don't need to get it down to a single commit - we can preserve authorship - but let's pare it down to a reasonable number of commits with clear commit messages explaining the contribution of each. Also, the |
Done going through the notes - I have removed .gitmodules, too! |
Awesome thanks. Ok, I'll give it a shot. |
Ok I give up. If I want to remove or squash commits I have to use On the bright side, we still have a working cuDNN v4 integration! |
CUDNN v4 upgrade. Optionally, CNMEM used for GPU pool allocation.
Great news, thanks Luke! From: Luke Yeager <notifications@github.commailto:notifications@github.com> Ok I give up. If I want to remove or squash commits I have to use git rebase -i, but I can't do that without removing all merge commits and attempting to turn the history into a linear stream of commits, which leads to lots of tricky merge commits. Not worth it. On the bright side, we still have a working cuDNN v4 integration! Reply to this email directly or view it on GitHubhttps://github.com//pull/42#issuecomment-149745429. This email message is for the sole use of the intended recipient(s) and may contain reply email and destroy all copies of the original message. |
endif | ||
endif | ||
|
||
PROJECT_DIR=$(PWD) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@borisfom why did you add this PROJECT_DIR
variable? It broke my debian packaging. Is it really necessary?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I needed so set paths to absolute, let’s discuss this.
From: Luke Yeager [mailto:notifications@github.com]
Sent: Wednesday, October 21, 2015 5:38 PM
To: NVIDIA/caffe
Cc: Boris Fomitchev
Subject: Re: [caffe] CUDNN v4 upgrade. Optionally, CNMEM used for GPU pool allocation. (#42)
In Makefilehttps://github.com//pull/42#discussion_r42700544:
@@ -7,12 +7,27 @@
$(error $ (CONFIG_FILE) not found. See $(CONFIG_FILE).example.)endif
include $(CONFIG_FILE)
+# Rectify input parameters
+ifeq ($(CPU_ONLY),1)
- USE_CUDNN=0
- USE_CNMEM=0
+endif
+ifeq ($(USE_CUDNN),1)
+# CNMEM is ON by default in CUDNN is ON
- ifeq ($(USE_CNMEM),)
- USE_CNMEM=1
- endif
+endif
+PROJECT_DIR=$(PWD)
@borisfomhttps://github.com/borisfom why did you add this PROJECT_DIR variable? It broke my debian packaging. Is it really necessary?
—
Reply to this email directly or view it on GitHubhttps://github.com//pull/42/files#r42700544.
This email message is for the sole use of the intended recipient(s) and may contain
confidential information. Any unauthorized review, use, disclosure or distribution
is prohibited. If you are not the intended recipient, please contact the sender by
Note: CNMEM(https://github.com/NVIDIA/cnmem.git) submodule added at caffe/cnmem.
Attention:
To populate added CNMEM submodule, do 'git submodule update'.
Alernatively, CNMEM can be pulled externally from https://github.com/NVIDIA/cnmem.git.
Caffe does not make any assumptions about actual CNMEM code location at the time.