Google GitGoogle Git
Sign in
android/platform/external/pytorch/24ece087c7
commit
24ece087c760a55ee1d5aef12228fd4a63a942de
[log]
author
Ahmed Taei <ataei@fb.com>
Fri Jul 21 17:24:43 2017 -0700
committer
Facebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri Jul 21 17:36:22 2017 -0700
tree
4a25d8213bc04553aae2ec4348283199b2a36b50
parent
804ebf7c41acdc4dd1e2181aa016017cb14cbb4b [diff]
Replace ReduceDimsOps math::Gemv with CUDA reduction kernel. 5.6x speed up.

Summary: This reduces runtime from 1.54757 ms/iter -> 0.273687 ms/iter for an 100 parallel reductions each of size 100000.

Reviewed By: akyrola

Differential Revision: D5471324

fbshipit-source-id: 626cabb8249fb4655275648fae2738cb739e1a72
1 file changed
tree: 4a25d8213bc04553aae2ec4348283199b2a36b50
  1. .travis/
  2. caffe/
  3. caffe2/
  4. cmake/
  5. docs/
  6. scripts/
  7. third_party/
  8. .Doxyfile
  9. .Doxyfile-c
  10. .Doxyfile-python
  11. .gitignore
  12. .gitmodules
  13. .travis.yml
  14. appveyor.yml
  15. CMakeLists.txt
  16. LICENSE
  17. Makefile
  18. PATENTS
  19. README.md
  20. release-notes.md
README.md

Caffe2

TravisCI Build Status Appveyor Build Status

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

News and Events

Caffe2 research award competition request for proposals

Questions and Feedback

Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.

Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.

License and Citation

Caffe2 is released under the BSD 2-Clause license.

Further Resources on Caffe2.ai