fix operator.cmake

test=develop

paddle/fluid/operators/lstm_cudnn_op.cc → paddle/fluid/operators/cudnn_lstm_op.cc

@@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
-#include "paddle/fluid/operators/lstm_cudnn_op.h"
+#include "paddle/fluid/operators/cudnn_lstm_op.h"
 #include <string>
 
 #ifdef PADDLE_WITH_CUDA
@@ -205,12 +205,12 @@ class CudnnLSTMGradOp : public framework::OperatorWithKernel {
 }  // namespace paddle
 
 namespace ops = paddle::operators;
-REGISTER_OPERATOR(lstm_cudnn, ops::CudnnLSTMOp, ops::CudnnLSTMOpMaker,
+REGISTER_OPERATOR(cudnn_lstm, ops::CudnnLSTMOp, ops::CudnnLSTMOpMaker,
                   paddle::framework::DefaultGradOpDescMaker<true>);
 REGISTER_OPERATOR(lstm_cudnn_grad, ops::CudnnLSTMGradOp);
 
 REGISTER_OP_CPU_KERNEL(
-    lstm_cudnn,
+    cudnn_lstm,
     ops::CudnnLSTMKernel<paddle::platform::CPUDeviceContext, float>);
 
 REGISTER_OP_CPU_KERNEL(

paddle/fluid/operators/lstm_cudnn_op.cu.cc → paddle/fluid/operators/cudnn_lstm_op.cu.cc

@@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
-#include "paddle/fluid/operators/lstm_cudnn_op.h"
+#include "paddle/fluid/operators/cudnn_lstm_op.h"
 #include "paddle/fluid/platform/cudnn_helper.h"
 
 namespace paddle {
@@ -487,8 +487,8 @@ class CudnnLSTMGPUGradKernel : public framework::OpKernel<T> {
 
 namespace ops = paddle::operators;
 REGISTER_OP_CUDA_KERNEL(
-    lstm_cudnn,
+    cudnn_lstm,
     ops::CudnnLSTMGPUKernel<paddle::platform::CUDADeviceContext, float>);
 REGISTER_OP_CUDA_KERNEL(
-    lstm_cudnn_grad,
+    cudnn_lstm_grad,
     ops::CudnnLSTMGPUGradKernel<paddle::platform::CUDADeviceContext, float>);