[CINN]Add InferSymbolicShape for tensor_to_array #69069
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
你的PR提交成功，感谢你对开源项目的贡献! |
ooooo-create
changed the title
ooooo-create
commented
Oct 31, 2024
|
Sorry to inform you that 3393252's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
| this->attributes().at("use_stack").dyn_cast<pir::BoolAttribute>().data(); | ||
| const auto &x_shape_or_data = | ||
| infer_context->GetShapeOrDataForValue(x()) | ||
| .dyn_cast<symbol::RankedTensorArrayShapeOrDataDimExprs>(); |
There was a problem hiding this comment.
语义上看 TensorToArrayOp 的输入参数应该是TensorShapeOrDataDimExprs吧
There was a problem hiding this comment.
第一个输入是 RankedTensorArrayShapeOrDataDimExprs
There was a problem hiding this comment.
好像只是反向算子，可能是求梯度的意思，求x的梯度
ooooo-create
changed the title
ooooo-create
commented
Nov 26, 2024
| const auto &x_shape_or_data = | ||
| infer_context->GetShapeOrDataForValue(x()) | ||
| .dyn_cast<symbol::RankedTensorArrayShapeOrDataDimExprs>(); | ||
| infer_context->SetShapeOrDataForValue(x_grad(), x_shape_or_data); |
There was a problem hiding this comment.
tensor_to_array 只有一个输出
zyfncg
approved these changes
Nov 26, 2024
Hongqing-work
approved these changes
Nov 26, 2024
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR Category
CINN
PR Types
Others
Description
Add InferSymbolicShape for select_output,PIR 下不支持，无需实现