saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2
History
Yan Xie 959a7138fd Support RowWiseSparseAdam on GPU (#34341) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34341

Implement RowWiseSparseAdam on CUDA

Reviewed By: xianjiec

Differential Revision: D20289209

fbshipit-source-id: a7a8a21bd18c1b9891f04f202d3ecaf183e30cad
2020-03-17 15:08:24 -07:00
..
contrib move emulation libraries to contrib (#34861) 2020-03-17 12:50:41 -07:00
core [Caffe2] Move more method implementations from tensor.h to tensor.cc (#34811) 2020-03-16 22:15:18 -07:00
cuda_rtc
db [Caffe2] Skip //caffe2/caffe2:caffe2_test_cpu -- 'DBSeekTest\.RocksDB' 2020-02-21 21:30:02 -08:00
distributed
experiments Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
ideep Enable mkldnn on windows (#31355) 2020-01-27 09:00:02 -08:00
image Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
mobile Fix spelling errors 2020-01-28 04:46:15 -08:00
mpi
observers preprocessor cleanup (#33957) 2020-03-02 13:37:19 -08:00
onnx Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
operators Revert D20461609: [caffe2] open source 2/4-bit SLS operators 2020-03-17 11:10:10 -07:00
opt Initial implementation of NNPI Int8FC op 2020-03-16 10:46:17 -07:00
perfkernels [caffe2] simplify caffe2 code with fbgemm handling block size 1 emb (#33774) 2020-02-27 14:45:28 -08:00
predictor Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
proto [DPER3][Shape Inference] Initial Shape Inference in DPER3 frontend (#33607) 2020-03-12 20:25:50 -07:00
python Support RowWiseSparseAdam on GPU (#34341) 2020-03-17 15:08:24 -07:00
quantization Fix missing header (#34762) 2020-03-15 00:19:42 -07:00
queue
serialize [jit] Add type tags to lists/dicts in pickle (#33255) 2020-03-10 19:17:01 -07:00
sgd Support RowWiseSparseAdam on GPU (#34341) 2020-03-17 15:08:24 -07:00
share [caffe2] simplify relative error expr (#32999) 2020-02-19 16:35:44 -08:00
test
transforms
utils [caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625) 2020-03-16 15:18:44 -07:00
video Fix compilation error when buildng with FFMPEG (#27589) 2020-02-13 11:23:48 -08:00
.clang-format
__init__.py
c2_aten_srcs.bzl
CMakeLists.txt Fix CMake Dev warning in caffe2/CMakeLists.txt (#34886) 2020-03-17 12:19:42 -07:00
README.md
release-notes.md
requirements.txt
VERSION_NUMBER

README.md

Caffe2

Jenkins 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.

Questions and Feedback

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

Further Resources on Caffe2.ai