projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 059c55f) | patch
Register CUDA kernels for caffe2 operators (#16691)
authorSebastian Messmer <messmer@fb.com>
Wed, 13 Feb 2019 00:47:53 +0000 (16:47 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 13 Feb 2019 01:24:01 +0000 (17:24 -0800)
commit9696fee635d85a93722ee66f36f17a78b2b7625b
tree0ffa9b82cf988d792e59107797139d86c5939a72tree | snapshot
parent059c55f8cc9b33dcbf78c71afad20cf640eae070commit | diff
Register CUDA kernels for caffe2 operators (#16691)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16691

Previous diffs already introduced a macro that registers caffe2 CPU kernels with c10.
This now also registers the CUDA kernels with it.

Reviewed By: bwasti

Differential Revision: D13901619

fbshipit-source-id: c15e5b7081ff10e5219af460779b88d6e091a6a6
caffe2/core/c10_operator.h diff | blob | history
caffe2/operators/layer_norm_op.cu diff | blob | history
caffe2/python/operator_test/layer_norm_op_test.py diff | blob | history
Domain: Machine Learning / ML Framework;