diff --git a/paddle/fluid/eager/eager_tensor.h b/paddle/fluid/eager/eager_tensor.h index 9303adb0acec556af36bfcdde68f4757ff62020c..d871a84dc224f60759649f331a980138ed18654f 100644 --- a/paddle/fluid/eager/eager_tensor.h +++ b/paddle/fluid/eager/eager_tensor.h @@ -21,6 +21,7 @@ // pten deps #include "paddle/pten/all.h" #include "paddle/pten/api/all.h" +#include "paddle/pten/api/lib/api_declare.h" #include "paddle/pten/api/lib/utils/tensor_utils.h" /** * This class is used by Eager mode for now. It's painful to do this in Eager diff --git a/paddle/fluid/eager/tests/data_structure_tests/eager_tensor_test.cc b/paddle/fluid/eager/tests/data_structure_tests/eager_tensor_test.cc index 60193cb1c88280e629e8661cd5a53be3f961a60b..a02f0bec456bf195d56ab3d47f7aa73c670b20a4 100644 --- a/paddle/fluid/eager/tests/data_structure_tests/eager_tensor_test.cc +++ b/paddle/fluid/eager/tests/data_structure_tests/eager_tensor_test.cc @@ -16,7 +16,6 @@ #include "gtest/gtest.h" #include "paddle/fluid/eager/eager_tensor.h" -#include "paddle/pten/api/lib/api_declare.h" #include "paddle/pten/api/lib/utils/allocator.h" namespace eager_test {