Skip to content

Support array attribute in stablehlo composite#6840

Merged
chunnienc merged 4 commits intomasterfrom
chunnienc/support-more-hlfb-attr-type
Mar 29, 2024
Merged

Support array attribute in stablehlo composite#6840
chunnienc merged 4 commits intomasterfrom
chunnienc/support-more-hlfb-attr-type

Conversation

@chunnienc
Copy link
Copy Markdown
Collaborator

Support translating list[int], list[float], list[bool] python value to array attribute in composite.

Internal reference: https://b.corp.google.com/issues/331634303

@chunnienc chunnienc requested review from GleasonK and lsy323 March 27, 2024 21:21
@chunnienc chunnienc added the stablehlo StableHLO related work label Mar 27, 2024
Copy link
Copy Markdown
Collaborator

@GleasonK GleasonK left a comment

Choose a reason for hiding this comment

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

LGTM. A few minor suggestions including error messages

Comment thread torch_xla/csrc/runtime/stablehlo_composite_helper.cc Outdated
Comment thread torch_xla/csrc/runtime/stablehlo_composite_helper.cc Outdated
Comment thread torch_xla/csrc/runtime/stablehlo_composite_helper.cc Outdated
Comment thread torch_xla/csrc/runtime/stablehlo_composite_helper.cc Outdated
@chunnienc chunnienc merged commit 7bbe9d7 into master Mar 29, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

stablehlo StableHLO related work

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants