diff --git a/paddle/operators/math/math_function.cu b/paddle/operators/math/math_function.cu
index 3a216993acf351cb6dcc5faf46c31f875f6878b1..255e480680499877ff599b96b8336a968cccbb34 100644
--- a/paddle/operators/math/math_function.cu
+++ b/paddle/operators/math/math_function.cu
@@ -12,6 +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/framework/data_type.h"
 #include "paddle/operators/math/math_function.h"
 
 namespace paddle {