Skip to content

Conversation

@masahi
Copy link
Member

@masahi masahi commented Oct 11, 2021

Currently, using dynamic inputs shape with gather op results in segfault at

ICHECK_GE(*indice_shape_i, 1);
, which is embarrassing. It turned out that even though gather is such a common op, it didn't support dynamic shapes. This PR fixed that. Thanks @interesaaat for reporting the bug.

please review @comaniac @mbrookhart @AndrewZhaoLuo

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@comaniac comaniac merged commit 5ad2f77 into apache:main Oct 11, 2021
@comaniac
Copy link
Contributor

Thanks @masahi

masahi added a commit to Laurawly/tvm-1 that referenced this pull request Oct 14, 2021
* support gather op dynamic input

* fix shape func and add test

* remove constness check

* fix shape func output rank

* restore check

Co-authored-by: masa <masa@pop-os.localdomain>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 7, 2022
* support gather op dynamic input

* fix shape func and add test

* remove constness check

* fix shape func output rank

* restore check

Co-authored-by: masa <masa@pop-os.localdomain>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
* support gather op dynamic input

* fix shape func and add test

* remove constness check

* fix shape func output rank

* restore check

Co-authored-by: masa <masa@pop-os.localdomain>
apivovarov pushed a commit to neo-ai/tvm that referenced this pull request Jun 12, 2022
* support gather op dynamic input

* fix shape func and add test

* remove constness check

* fix shape func output rank

* restore check

Co-authored-by: masa <masa@pop-os.localdomain>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants