OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
Tensors and Dynamic neural networks in Python with strong GPU acceleration pytorch.org
2,337 Commits 8,304 Branches 150 Tags 1.5 GiB
Python 60.1%
C++ 32.1%
Cuda 2.9%
C 1.4%
Objective-C++ 1.1%
Other 2.1%
bb08f261f1
Go to file
Aapo Kyrola bb08f261f1 EnsureDense/SparseToDense for CUDA 
Summary: Make CUDA version of SparseToDense, register EnsureDense (which is trivial) on CUDA. Need to use atomics because indices can be duplicated. We can later add an option to inform if the indices are unique, and use faster path then.

Reviewed By: jhcross

Differential Revision: D5750893

fbshipit-source-id: 005d1675b127a571aac8474fca62d9633f0c7bff
2017-09-01 09:33:05 -07:00
.travis fix travis build 2017-08-31 23:35:20 -07:00
caffe/proto cmake: relative paths for install() 2017-08-22 09:52:09 -07:00
caffe2 EnsureDense/SparseToDense for CUDA 2017-09-01 09:33:05 -07:00
cmake protoc: only disable in watch os mode 2017-09-01 00:46:51 -07:00
conda Conda packaging for caffe2. 2017-08-08 21:55:30 -07:00
docker Docker move 2017-08-09 13:54:17 -07:00
docs Remove map() and filter() in favor of comprehensions 2017-05-30 15:32:58 -07:00
scripts Allow caffe2 to detect if cuda lib has been linked, and also fix oss build error. 2017-08-23 18:41:15 -07:00
third_party Update gloo dependency 2017-08-25 12:01:23 -07:00
.Doxyfile Fix a few typos and grammars in comment 2017-06-14 18:22:39 -07:00
.Doxyfile-c changed doxygen config to new docs path (#311) 2017-04-19 11:49:59 -07:00
.Doxyfile-python changed doxygen config to new docs path (#311) 2017-04-19 11:49:59 -07:00
.gitignore Added Ninja generator support on Windows 2017-07-26 00:32:20 -07:00
.gitmodules Add build support for opengl and latest nnpack. 2017-08-09 00:31:53 -07:00
.travis.yml Bump travis osx to 8.3.3 2017-08-09 01:21:39 -07:00
appveyor.yml Fix a few typos and grammars in comment 2017-06-14 18:22:39 -07:00
cmake_uninstall.cmake.in add in make uninstall for cmake 2017-08-08 22:10:07 -07:00
CMakeLists.txt Include Caffe2 headers before anything else 2017-08-30 10:22:06 -07:00
LICENSE fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
Makefile Fix python support problems caused by building script errors. 2017-04-24 17:17:21 -07:00
PATENTS Create PATENTS 2017-04-07 13:46:29 -07:00
README.md updated research proposal link 2017-07-24 08:56:02 -07:00
release-notes.md launch updates (#309) 2017-04-19 11:40:51 -07:00

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