Skip to content

[ShapeFunc] Handle weights in shape func#6912

Merged
anijain2305 merged 2 commits intoapache:mainfrom
anijain2305:shape_func_cont
Nov 14, 2020
Merged

[ShapeFunc] Handle weights in shape func#6912
anijain2305 merged 2 commits intoapache:mainfrom
anijain2305:shape_func_cont

Conversation

@anijain2305
Copy link
Copy Markdown
Contributor

When compiling a dynamic model with TRT, the primitive function has constants already bound in the graph. TVM codegen typically lifts those into variables in ANF pass, so we never see this problem in TVM codegen.

@icemelon9 @zhiics @codeislife99 @rohanmukh @trevor-m

Copy link
Copy Markdown
Member

@zhiics zhiics left a comment

Choose a reason for hiding this comment

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

LGTM. @icemelon9 PTAL

Copy link
Copy Markdown
Member

@icemelon icemelon left a comment

Choose a reason for hiding this comment

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

LGTM. just a minor comment.

"shape_const", topi::kBroadcast);
scalars_.push_back(value);
return {value};
} else {
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

No need for else since you already return.

@anijain2305 anijain2305 merged commit e009188 into apache:main Nov 14, 2020
@anijain2305
Copy link
Copy Markdown
Contributor Author

Thanks @zhiics @icemelon9 This is merged.

TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Nov 20, 2020
* [ShapeFunc] Handle weights in shape func
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
* [ShapeFunc] Handle weights in shape func

* Comments
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
* [ShapeFunc] Handle weights in shape func

* Comments
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
* [ShapeFunc] Handle weights in shape func

* Comments
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.

3 participants