Skip to content

[autoparallel] bypass MetaInfo when unavailable and modify BCAST_FUNC_OP metainfo#2293

Merged
Cypher30 merged 11 commits intohpcaitech:debug/ckpt-autoparallelfrom
Cypher30:debug/ckpt-autoparallel
Jan 3, 2023
Merged

[autoparallel] bypass MetaInfo when unavailable and modify BCAST_FUNC_OP metainfo#2293
Cypher30 merged 11 commits intohpcaitech:debug/ckpt-autoparallelfrom
Cypher30:debug/ckpt-autoparallel

Conversation

@Cypher30
Copy link
Copy Markdown
Contributor

@Cypher30 Cypher30 commented Jan 3, 2023

What's New?

In this PR I modify the meta information for BCAST_FUNC_OP to fix the case like torch.true_divide that only has one input tensor. I also bypass MetaInfo when the operation is not patched in the meta registry.

@Cypher30 Cypher30 merged commit b904748 into hpcaitech:debug/ckpt-autoparallel Jan 3, 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.

1 participant