Fix custom device compile error caused by dist marco changing (Paddle…

…Paddle#56760) * fix custom device errro by dist * polish details
BeingGod · Sep 9, 2023 · bc12e68 · bc12e68
1 parent 5655892
commit bc12e68
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/paddle/phi/core/meta_tensor.cc b/paddle/phi/core/meta_tensor.cc
@@ -17,6 +17,7 @@ limitations under the License. */
 #include "glog/logging.h"
 
 #include "paddle/phi/core/dense_tensor.h"
+#include "paddle/phi/core/distributed/auto_parallel/dist_tensor.h"
 #include "paddle/phi/core/enforce.h"
 #include "paddle/phi/core/selected_rows.h"
 #include "paddle/phi/core/string_tensor.h"

diff --git a/paddle/phi/core/tensor_utils.h b/paddle/phi/core/tensor_utils.h
@@ -16,7 +16,6 @@ limitations under the License. */
 
 #include "paddle/phi/core/dense_tensor.h"
 #include "paddle/phi/core/device_context.h"
-#include "paddle/phi/core/distributed/auto_parallel/dist_tensor.h"
 #include "paddle/phi/core/selected_rows.h"
 #include "paddle/phi/core/sparse_coo_tensor.h"
 #include "paddle/phi/core/sparse_csr_tensor.h"