CARVIEW |
Navigation Menu
-
Notifications
You must be signed in to change notification settings - Fork 5.8k
[CINN] Add RepeatInterleaveWithTensorIndex && Solve #68550
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
33cb4cc
to
ab6aced
Compare
Sorry to inform you that ab6aced's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
Outdated
Show resolved
Hide resolved
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
Outdated
Show resolved
Hide resolved
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
Outdated
Show resolved
Hide resolved
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
Outdated
Show resolved
Hide resolved
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
Outdated
Show resolved
Hide resolved
std::vector<symbol::DimExpr> out_sym_shape; | ||
for (int i = 0; i < x_rank; i++) { | ||
if (i == axis) { | ||
out_sym_shape.push_back(in_dims_sym.at(i) * repeat_times_sym); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
repeats参数可以是个tensor,所以repeats的size应该跟in_dims_sym.at(i)相等,且推导出的维度应该是repeat 内的元素和。如果repeats是个值的话,可以用 in_dims_sym.at(i) * repeat
@@ -211,11 +211,50 @@ void InferSymbolicShapeContext::AddEqualCstr(const symbol::DimExpr& lhs, | |||
constraints_manager_.AddEqCstr(lhs, rhs); | |||
} | |||
|
|||
void InferSymbolicShapeContext::AddEqualCstr( | |||
const std::vector<symbol::DimExpr>& lhs, | |||
const std::vector<symbol::DimExpr>& rhs) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里语义有些不明确,补充个注释说明两个Vector添加EqualCstr的含义
Sorry to inform you that 374e2c1's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
Sorry to inform you that 6c8403a's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
@@ -57,7 +57,7 @@ def init_dtype_type(self): | |||
self.index_size = self.x_shape[self.dim] | |||
|
|||
def test_check_output(self): | |||
self.check_output(check_pir=True) | |||
self.check_output(check_pir=True, check_symbol_infer=False) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个单测为什么关掉了?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个算子的Repeats参数是做为输入传进来的,Get的时候产生了新符号
PR Category
CINN
PR Types
New features
Description