Skip to content

Set TOpPattern=kOpaque for scatter_nd#7464

Merged
tqchen merged 1 commit intoapache:mainfrom
apivovarov:scatter_nd
Feb 18, 2021
Merged

Set TOpPattern=kOpaque for scatter_nd#7464
tqchen merged 1 commit intoapache:mainfrom
apivovarov:scatter_nd

Conversation

@apivovarov
Copy link
Copy Markdown
Contributor

@apivovarov apivovarov commented Feb 17, 2021

Set TOpPattern=kOpaque for scatter_nd op to fix the issue described here
https://discuss.tvm.apache.org/t/scatter-nd-and-concatenate-fails/9138
@masahi @anijain2305 @tkonolige @haan

Copy link
Copy Markdown
Contributor

@tkonolige tkonolige left a comment

Choose a reason for hiding this comment

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

I'm not very familiar with fusion. I should have guess that this would cause an issue.

Thanks @apivovarov!

@anijain2305
Copy link
Copy Markdown
Contributor

@apivovarov Can you please add the failing test from the discuss post?

And also please add a comment on the lines of "scatter_nd operator has extern schedules for CPU and GPU devices. Fusing extern schedules with injective schedules leads to errors. So, converting the scatter_nd to Opaque to prevent compilation failures"

@tqchen tqchen merged commit 84c4b15 into apache:main Feb 18, 2021
@tqchen
Copy link
Copy Markdown
Member

tqchen commented Feb 18, 2021

Thanks @apivovarov for catching the problem! Thanks @anijain2305 @tkonolige

masahi pushed a commit to masahi/tvm that referenced this pull request Feb 22, 2021
Lokiiiiii pushed a commit to Lokiiiiii/tvm that referenced this pull request Mar 2, 2021
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants