Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
P
Paddle
提交 46153c78 编写于 作者: L liangjianzhong
浏览文件
操作

shape int64

上级 f314b562
显示空白变更内容
内联 并排
Showing with 2 addition and 2 deletion
paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.cc
浏览文件 @ 46153c78
...@@ -165,7 +165,7 @@ std::vector<DistTensorSpec> MatmulSPMDRule::InferForward( ...@@ -165,7 +165,7 @@ std::vector<DistTensorSpec> MatmulSPMDRule::InferForward(
TensorDistAttr GetInferedDistAttr( TensorDistAttr GetInferedDistAttr(
const TensorDistAttr& origin_dist_attr, const TensorDistAttr& origin_dist_attr,
const std::vector<int>& shape, const std::vector<int64_t>& shape,
const std::string& tensor_axis, const std::string& tensor_axis,
const std::unordered_map<std::string, int64_t>& axis_to_dim_map) { const std::unordered_map<std::string, int64_t>& axis_to_dim_map) {
TensorDistAttr dist_attr_ = CopyTensorDistAttrForOutput(origin_dist_attr); TensorDistAttr dist_attr_ = CopyTensorDistAttrForOutput(origin_dist_attr);
......
paddle/fluid/distributed/auto_parallel/spmd_rules/matmul_spmd_rule.h
浏览文件 @ 46153c78
...@@ -27,7 +27,7 @@ namespace auto_parallel { ...@@ -27,7 +27,7 @@ namespace auto_parallel {
TensorDistAttr GetInferedDistAttr( TensorDistAttr GetInferedDistAttr(
const TensorDistAttr& origin_dist_attr, const TensorDistAttr& origin_dist_attr,
const std::vector<int>& shape, const std::vector<int64_t>& shape,
const std::string& tensor_axis, const std::string& tensor_axis,
const std::unordered_map<std::string, int64_t>& axis_to_dim_map); const std::unordered_map<std::string, int64_t>& axis_to_dim_map);
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或