Jiawei Wu comments

Results 7 comments of


                                            Jiawei Wu

[MHLO] Add configurable op decomposition rules

> Can we add some file check unit tests for the pass? Done.

[MHLO] Add configurable op decomposition rules

> I have some specific opinions about how we want to do this -- I am on vacation the rest of this week, but after #1165 lands this was the...

[MHLO] Add configurable op decomposition rules

There is a better choice in PR #1242 . This PR is useless and closed.

emitted aten.norm.ScalarOpt_dim

Any progress with this PR?

[FxImporter]: support parsing nodes which return list

> Thanks for the contribution. If you wouldn't mind, could you add a test? I know the testing in tree is thin for this (we have a bunch of tests...

[FxImporter]: support parsing nodes which return list

> Thank you for the contribution. Since this stuff gets a bit twisty, do you mind if I take a bit of time and poke at this so that I...

[torch-frontend] add stablehlo IRs for Mixtral model.

Update at 2024.05.31. We add stablehlo IR of a **whole Mixtral 8x7B model**. Note, to save compilation time and memory consumption, we convert the large weights into splat DenseElementsAttrs. See...