cuda implementation for PackSegment to support presence mask (#14635)
authorYan Zhu <yzhu@fb.com>
Sat, 1 Dec 2018 00:50:37 +0000 (16:50 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 1 Dec 2018 00:54:10 +0000 (16:54 -0800)
commitaeb38cfcea0debfb44bd06f06e2d826e637968db
tree197543b29f0c17ecc7a85afce1c14e7170bd2924
parent1d464d7f3ea19b5698522fe28bd9adf29be1585d
cuda implementation for PackSegment to support presence mask (#14635)

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

as title

Reviewed By: enosair

Differential Revision: D13254097

fbshipit-source-id: b9f40109e2889907c925f9a4df9da14f67f45f38
caffe2/operators/pack_segments.cu
caffe2/python/operator_test/pack_ops_test.py