Skip to content

[Disco] Support ShapeTuple in Disco Protocol#15634

Merged
junrushao merged 1 commit intoapache:unityfrom
junrushao:feature/2023-08-28/support-str
Aug 28, 2023
Merged

[Disco] Support ShapeTuple in Disco Protocol#15634
junrushao merged 1 commit intoapache:unityfrom
junrushao:feature/2023-08-28/support-str

Conversation

@junrushao
Copy link
Copy Markdown
Member

ShapeTuple is an essential item used in Relax stack as the runtime representation of shapes. It is also used as a current workaround to represent integers (1-d shape) given standalone non-constant integers are currently absent in Relax.

This PR introduces formal support for ShapeTuple in Disco's communication protocol based on the recent enhancement of TVM RPC system to support TVM Objects: #15631.

ShapeTuple is an essential item used in Relax stack as the runtime
representation of shapes. It is also used as a current workaround to
represent integers (1-d shape) given standalone non-constant integers
are currently absent in Relax.

This PR introduces formal support for ShapeTuple in Disco's
communication protocol based on the recent enhancement of TVM RPC system
to support TVM Objects: apache#15631.
@junrushao junrushao marked this pull request as ready for review August 28, 2023 07:33
@junrushao
Copy link
Copy Markdown
Member Author

CC @jinhongyii @LeshengJin @tqchen

Copy link
Copy Markdown
Contributor

@LeshengJin LeshengJin left a comment

Choose a reason for hiding this comment

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

LGTM! Thanks @junrushao

Comment thread src/runtime/disco/threaded_session.cc
@junrushao junrushao merged commit 0230c77 into apache:unity Aug 28, 2023
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.

4 participants