shape int64

paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.cc

@@ -165,7 +165,7 @@ std::vector<DistTensorSpec> MatmulSPMDRule::InferForward(
 
 TensorDistAttr GetInferedDistAttr(
     const TensorDistAttr& origin_dist_attr,
-    const std::vector<int>& shape,
+    const std::vector<int64_t>& shape,
     const std::string& tensor_axis,
     const std::unordered_map<std::string, int64_t>& axis_to_dim_map) {
   TensorDistAttr dist_attr_ = CopyTensorDistAttrForOutput(origin_dist_attr);

paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.h

@@ -27,7 +27,7 @@ namespace auto_parallel {
 
 TensorDistAttr GetInferedDistAttr(
     const TensorDistAttr& origin_dist_attr,
-    const std::vector<int>& shape,
+    const std::vector<int64_t>& shape,
     const std::string& tensor_axis,
     const std::unordered_map<std::string, int64_t>& axis_to_dim_map);