Joel Berkeley

Results 103 comments of Joel Berkeley

> Uniqueness of keys and dictionary indexing remains unchanged and is left as an implementation choice of the library writer. Indeed, I mainly want to check if this functionality will...

closing as I think I've missed sth, may reopen

OK thanks. I'll need to do some reading to understand (I'm unfamiliar with "emit", and "pass") Is this all in Enzyme-JAX repo or is some in Enzyme repo?

thanks, and are there C++ apis for this? Looking at the Julia implementation "emit" seems to defer to the MLIR C++ API. I'm not sure what running an optimization pass...

thanks. I did look at that function, but it's not available in the public API of Enzyme-JAX. I might just copy its intention. I more meant APIs for building `enzyme.autodiff`...

Where can I find the relevant dialects and passes? I tried ``` registry.insert() mlir::stablehlo::registerAllDialects(registry) ``` and ``` mlir::enzyme::createDifferentiatePass(); ``` but that's not working. I don't know if that's the problem....

BTW once I work out how to do this, I might make a small C++ library for differentiating StableHLO, one that's not specific to any particular frontend

> isn't that just the `enzymexla-interpreter`? what's that?

I'm differentiating a C++ `mlir::ModuleOp`. The StableHLO is generated at runtime