saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
Tensors and Dynamic neural networks in Python with strong GPU acceleration
336 commits 5021 branches 1206 tags 1 GiB
Python 56.8%
C++ 35%
Cuda 3%
C 1.5%
Objective-C++ 1.1%
Other 2.3%
Find a file
Martin Raison da72658fa8 sparsehash-based implementation of UniqueOp 
Summary:
Faster implementation of UniqueOp using google::dense_hash_map, as suggested by dzhulgakov. I haven't benchmarked it precisely but early measurements with my workflow show a significant speed bump (this operation went from using 20% of overall CPU time down to 7%).

I gated the implementation using the "engine" feature, to avoid adding sparsehash as a dependency to caffe2.

Reviewed By: dzhulgakov

Differential Revision: D4219768

fbshipit-source-id: 2f142981e772105b42fffa24afb199ef816f8e0c
2016-11-29 15:18:39 -08:00
caffe/proto more build updates: 2016-08-02 23:28:23 -07:00
caffe2 sparsehash-based implementation of UniqueOp 2016-11-29 15:18:39 -08:00
docs chunky sync 2016-05-13 14:43:48 -07:00
third_party nervana build files 2016-10-07 18:37:26 -07:00
.Doxyfile Adding optional Eigen code. Added a switch USE_SYSTEM_EIGEN in Env. Misc changes. 2015-10-18 16:55:24 -07:00
.gitignore Adding optional Eigen code. Added a switch USE_SYSTEM_EIGEN in Env. Misc changes. 2015-10-18 16:55:24 -07:00
.gitmodules nervana build files 2016-10-07 18:37:26 -07:00
build.py nervana build files 2016-10-07 18:37:26 -07:00
build_android.py fbsync 2016-10-07 15:47:52 -07:00
build_android_prepare.py fbsync 2016-10-07 15:47:52 -07:00
LICENSE fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00
Makefile more build updates: 2016-08-02 23:28:23 -07:00
README.md removed deprecated readme info 2016-11-18 17:03:20 -08:00

README.md

Caffe2

Caffe2 is a deep learning framework made with expression, speed, and modularity in mind. It is an experimental refactoring of Caffe, and allows a more flexible way to organize computation.

Read the installation instructions for installation details.

License and Citation

Caffe2 is released under the BSD 2-Clause license.