Add/fallback some operators for mkl-dnn (#11696)
authorCheng,Penghui <penghui.cheng@intel.com>
Fri, 11 Jan 2019 20:48:57 +0000 (12:48 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 11 Jan 2019 20:53:06 +0000 (12:53 -0800)
commit926e718d5fe129f67a10eef7ef8ce754b25c1e1e
tree759265d7d43a1a580ebfe794b7bb54f1db848392
parent96ea2594d882fc177a6730c42cd3c266f9bb2a67
Add/fallback some operators for mkl-dnn (#11696)

Summary:
Implementation LeakyRelu operator for mkl-dnn,the speed-up of a single operation is up to 10X on BDW.
Implementation rashape operator for mkl-dnn,it will resolve occasionally crash issue which use fallback reshape operator.
Implementation CreateBlobQueue and SafeEnqueueBlobs operators,it will resolve crash issue which use fallback operators.
Fallback CreateBlobsQueueDBOp,TensorProtosDBInput,CloseBlobsQueue operators.
Implement adam operator for mkl-dnn,the speed-up of a single operator is up to 6X on BDW.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11696

Reviewed By: yinghai

Differential Revision: D10100438

Pulled By: wesolwsk

fbshipit-source-id: 0b6e06897cc11e0a8e349d80a870b1e72e47f10d
caffe2/ideep/operators/adam_op.cc [new file with mode: 0644]
caffe2/ideep/operators/operator_fallback_ideep.cc
caffe2/ideep/operators/operator_fallback_ideep.h
caffe2/ideep/operators/queue_ops.cc [new file with mode: 0644]
caffe2/ideep/operators/relu_op.cc
caffe2/ideep/operators/reshape_op.cc [new file with mode: 0644]
caffe2/python/ideep/adam_op_test.py [new file with mode: 0644]
caffe2/python/ideep/blobs_queue_db_test.py [new file with mode: 0644]
caffe2/python/ideep/leaky_relu_op_test.py [new file with mode: 0644]
caffe2/python/ideep/reshape_ops_test.py [new file with mode: 0644]
caffe2/queue/blobs_queue_db.cc