Skip to content

Add explicit error message for def infer_size(a, b): that specificy that non broadcast path was picked due to unbacked existing in both inputs. #152656

@laithsakka

Description

@laithsakka

follow up task for my comment on this PR
https://github.com/pytorch/pytorch/pull/152146/files

you can land it as for the future i might revisit this to make the code more understandable

basically first we want to check if there is broadcasting using guard_or_none.
if any of them did not return none we are done.
if both return non i would want to add an explicit extra message to the torch_check that says we have assumed this path because both sizeA == sizeB are unbacked

not action required from you at this moment. i will file issue for this

cc @chauhang @penguinwu @ezyang @bobrenjc93

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: dynamic shapesoncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions