pytorch
16e3f158 - Add function to port FX minified graph to HLO via StableHLO (#109084)

Commit
1 year ago
Add function to port FX minified graph to HLO via StableHLO (#109084) If `XLA_HLO_DEBUG` flag is enabled, generated a minified HLO graph when using the minifier. This function enables HLO minification support by porting the minified FX graph to StableHLO via the `save_torch_model_as_stablehlo` function. This allows users to port the minified graph to compilers that are not compatible with TorchDynamo/Inductor workflow and use XLA instead. The purpose of this PR is to help XLA users debug accuracy and compilation errors. It will also be helpful for existing TorchDynamo/XLA workflow on `torchxla_trace_once` backend as well. Fixes [#5461](https://github.com/pytorch/xla/issues/5461) in Torch XLA repo. CC @GleasonK @qihqi Pull Request resolved: https://github.com/pytorch/pytorch/pull/109084 Approved by: https://github.com/anijain2305
Author
Committer
Parents
Loading