William Moses

HLO Canonicalizations Todo list

31

To mark which ones we see worth doing, are doing / need to do cc @ivanradanov @ftynse - [x] iota reshape (becomes single iota) ``` %195 = stablehlo.iota dim =...

Correctly adapt to ABI changes to EnzymeCore

3

cc @michel2323 cc @jlk9

Ideally should also be forward/backported to other relevant ROCM versions. HIPRtc currently does not correctly support LLVM plugins. Specifically, symbols defined within LLVM and/or clang are not available by any...

Create a flag to use Enzyme as the AD in training/etc.

6

### Motivation and description Now that all the internal Flux tests pass, we should start setting up for integration. Having such a flag would make it easier for myself and...