diff --git a/paddle/framework/pybind.cc b/paddle/framework/pybind.cc
index cc2bad6234fd77c631617dc1cedef0465bbe7840..a955191e982e10372de861aec2c0af7b747fe828 100644
--- a/paddle/framework/pybind.cc
+++ b/paddle/framework/pybind.cc
@@ -29,7 +29,7 @@ limitations under the License. */
 namespace py = pybind11;
 
 USE_OP(add_two);
-USE_OP_CPU(onehot_cross_entropy);
+USE_CPU_OP(onehot_cross_entropy);
 USE_OP_ITSELF(fc);
 USE_OP(sgd);
 USE_OP(mul);