projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e596d23) | patch
FP16MomentumSGDUpdate Op fix and enable for ROCm (#15150)
authorrohithkrn <rohith.nallamaddi@gmail.com>
Sat, 15 Dec 2018 00:31:34 +0000 (16:31 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 15 Dec 2018 00:33:45 +0000 (16:33 -0800)
commit763b9954f3fbe0b2058eee9fe8055dfdc58ee615
tree44dd4ac008bcf9beb956ab286df615bccccabd66tree | snapshot
parente596d2313761ce0c93f2f6d379344aa22620b06bcommit | diff
FP16MomentumSGDUpdate Op fix and enable for ROCm (#15150)

Summary:
1. Fix a bug in FP16MomentumSGDUpdate operator
2. Enable operator for ROCm
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15150

Differential Revision: D13473145

Pulled By: bddppq

fbshipit-source-id: 4c5c5f30cb9bba658e3639dbe193fa08a304d306
caffe2/python/operator_test/momentum_sgd_test.py diff | blob | history
caffe2/sgd/fp16_momentum_sgd_op.cu diff | blob | history
Domain: Machine Learning / ML Framework;