Optimize relu op on GPU (#18506)
authorXiaomeng Yang <yangxm@fb.com>
Fri, 29 Mar 2019 07:20:25 +0000 (00:20 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 29 Mar 2019 07:23:24 +0000 (00:23 -0700)
commitc21e763cd6723a50d5acb50fed0f54252cf316ff
treee56f18b0e3a932dcf4dadac9b56a1abca7494b4f
parenta5a1c9a171874971b7ddf276c0332c064e3fffec
Optimize relu op on GPU (#18506)

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

Optimize relu op on GPU

Reviewed By: houseroad

Differential Revision: D14633171

fbshipit-source-id: bd3afa9a0bae1325d32ad4153736a0c7ecb0ec64
caffe2/operators/relu_op.cu
caffe2/operators/relu_op_cudnn.cc [deleted file]