From 46153c788318e073eb2cc5935faf6d063a5947e2 Mon Sep 17 00:00:00 2001
From: liangjianzhong <liangjianzhong@baidu.com>
Date: Tue, 23 May 2023 16:45:55 +0800
Subject: [PATCH] shape int64

---
 .../distributed/auto_parallel/spmd_rules/matmul_spmd_rule.cc    | 2 +-
 .../distributed/auto_parallel/spmd_rules/matmul_spmd_rule.h     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.cc b/paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.cc
index 9d6a973323e..74be7c143ef 100644
--- a/paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.cc
+++ b/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);
diff --git a/paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.h b/paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.h
index 89677e8e4e2..65b0600eae5 100644
--- a/paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.h
+++ b/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);
 
-- 
GitLab