tree d0768b8cad0f83aa764349f2324b7c6b0823d2db
parent 21c6cd2649e6d060c95af90a715eab3a42e83d95
author pstarkcdpr <paul.stark@cdprojektred.com> 1785525580 -0700
committer GitHub <noreply@github.com> 1785525580 +0200
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJqbPVMCRC1aQ7uu5UhlAAAp4AQABwQD4X/17oVKIKw703ccxAM
 or8q6j2DG+5YlcfykQ4RytpBMg3J5KETtqwAd7bPgKpb8zfsrT4dkiFEhmwdN3Qp
 YCyzyhCF7ELDLWvGCM4fxvdr0jYtuhBj50JtsjYoeWtimZFsKRHDn6nzorsq5FUN
 /IiHxn1qq8SXnT4h+ATZOki2iv9CswuEJE7TI/g1X0uPbpnGWZcH98m906gaUOE5
 MI9h1AbE9onI8bsFLqhw6OygJKSqE9C4wGgc/Cfl0npPkrzqMoTQJIh8iI+Ti1//
 nABSiost9fr8sv70spkJhE0z6J2wgsAMJG23fbeP6EKkeujjRcVg1WtXWjQjCFWe
 i3EVnijj07aVibW3FCpVhq6d3FzLQr/whncpiYQu7tdfrtJgPSJERlT2EEy+CGDS
 UgvWOQToXvA64Ndj2HqQD+OGDa4oRgLh6nSW668mYsRPkThFbJci93dajoKfgk7w
 8wtrCyKCtGolKlChysmKDOUft15MCjC3kOCIoCuu58InZuSdrOzhZT0sMcEM9iv0
 rNbn7piQ+SOu55yRkC/sW7MlIGhDcw95Bv2C+KRbLoINfuOpCAL9DOyN5VWNZMcc
 iQCYWOhicai49N5LGETNikYy8dnMlAGUV8wjT2djKLnF39LDa+SCw61WwWOrnO20
 zTk8kOTfjLXEkzkiTZWz
 =2y5D
 -----END PGP SIGNATURE-----
 

[StableHLO] Legalize composite ops to calls in the input pipeline (#24777)

The StableHLO input conversion pipeline had no handling for
`stablehlo.composite`, so any composite reaching the conversion to
Linalg failed with:
```
  error: failed to legalize operation 'stablehlo.composite' that was explicitly marked illegal
```

JAX emits composites for some CHLO operations, e.g.
```
  %0 = stablehlo.composite "chlo.acos" %arg0
      {decomposition = @chlo.acos.impl, version = 1 : i32}
      : (tensor<8x32xf32>) -> tensor<8x32xf32>
```
which made such models uncompilable.

Run `stablehlo-legalize-composite-to-call` early in the pipeline, right
after deserialization. It replaces each composite with a call to the
function holding its decomposition, which the inliner later in the
pipeline folds away, so the remaining conversions only ever see the
decomposed form. The pass is given an explicit empty exception list to
document where composites that should be preserved as a unit can be
opted out.

Signed-off-by: Paul Stark <paul.stark@cdprojektred.com>
Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>