diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index dc1fec12e..d14d084b8 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -9,7 +9,7 @@ http_archive( urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz".format(commit = NSYNC_COMMIT)], ) -ENZYMEXLA_COMMIT = "23f56612ecff51082166fe7ad9b80c9058742383" +ENZYMEXLA_COMMIT = "e70bd0a5c670937606972aa7bc0bf3990948730e" ENZYMEXLA_SHA256 = "" http_archive( @@ -129,9 +129,7 @@ http_archive( patches = ["@enzyme_ad//:patches/jax.patch"], ) -# load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256") -XLA_COMMIT = "610bf0cda8acc144fc843a6580282c39fd537824" -XLA_SHA256 = "" +load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256") http_archive( name = "xla", diff --git a/src/Compiler.jl b/src/Compiler.jl index b6fd0a7c1..4549bbd5f 100644 --- a/src/Compiler.jl +++ b/src/Compiler.jl @@ -709,7 +709,7 @@ function compile_mlir!( raise elseif raise # Raise enabled but use default passes - "canonicalize,llvm-to-memref-access,canonicalize,convert-llvm-to-cf,canonicalize,enzyme-lift-cf-to-scf,canonicalize,func.func(canonicalize-loops),canonicalize-scf-for,canonicalize,affine-cfg,canonicalize,func.func(canonicalize-loops),canonicalize,llvm-to-affine-access,canonicalize,delinearize-indexing,canonicalize,raise-affine-to-stablehlo,arith-raise{stablehlo=true}," * + "canonicalize,llvm-to-memref-access,canonicalize,convert-llvm-to-cf,canonicalize,enzyme-lift-cf-to-scf,canonicalize,func.func(canonicalize-loops),canonicalize-scf-for,canonicalize,affine-cfg,canonicalize,func.func(canonicalize-loops),canonicalize,llvm-to-affine-access,canonicalize,delinearize-indexing,canonicalize,affine-cfg,canonicalize,raise-affine-to-stablehlo,arith-raise{stablehlo=true}," * opt_passes2 else "canonicalize"