fix slice cpu bug (#68948)

phlrain · web-flow · commit 59dba7a7b8b3 · 2024-10-26T23:48:36.000+08:00
diff --git a/paddle/cinn/hlir/op/transform.cc b/paddle/cinn/hlir/op/transform.cc
@@ -1484,7 +1484,8 @@ std::shared_ptr<OpStrategy> StrategyForSliceSymbolic(
       return value.value();
     }
     if (attrs.attr_store.find("starts") != attrs.attr_store.end()) {
-      return ToCinnExprs(GetIntVectorFromAttr(attrs.attr_store.at("starts")));
+      return ToCinnExprs(
+          GetIntVectorFromAttr<int64_t>(attrs.attr_store.at("starts")));
     } else {
       PADDLE_THROW(::common::errors::InvalidArgument(
           "The Slice op doesn't find [starts] attribute!"));
@@ -1500,7 +1501,8 @@ std::shared_ptr<OpStrategy> StrategyForSliceSymbolic(
       return value.value();
     }
     if (attrs.attr_store.find("ends") != attrs.attr_store.end()) {
-      return ToCinnExprs(GetIntVectorFromAttr(attrs.attr_store.at("ends")));
+      return ToCinnExprs(
+          GetIntVectorFromAttr<int64_t>(attrs.attr_store.at("ends")));
     } else {
       PADDLE_THROW(::common::errors::InvalidArgument(
           "The Slice op doesn't find [ends] attribute!"));
diff --git a/paddle/cinn/hlir/pe/transform.cc b/paddle/cinn/hlir/pe/transform.cc
@@ -1337,7 +1337,7 @@ ir::Tensor SliceSymbolic(const ir::Tensor& A,
   }
 
   std::vector<Expr> new_starts = starts;
-  std::vector<int> axes;
+  std::vector<int64_t> axes;
   std::transform(const_axes.begin(),
                  const_axes.end(),
                  std::back_inserter(axes),

Original file line number	Diff line number	Diff line change
`@@ -1337,7 +1337,7 @@ ir::Tensor SliceSymbolic(const ir::Tensor& A,`
`1337`	`1337`	`}`
`1338`	`1338`
`1339`	`1339`	`std::vector<Expr> new_starts = starts;`
`1340`		`- std::vector<int> axes;`
	`1340`	`+ std::vector<int64_t> axes;`
`1341`	`1341`	`std::transform(const_axes.begin(),`
`1342`	`1342`	`const_axes.end(),`
`1343`	`1343`	`std::back_inserter(axes),`