wip

joelberkeley · Jan 4, 2025 · d50956a · d50956a
1 parent 33843fc
commit d50956a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spidr/backend/src/Enzyme-JAX/src/enzyme_ad/jax/Passes/Passes.cpp b/spidr/backend/src/Enzyme-JAX/src/enzyme_ad/jax/Passes/Passes.cpp
@@ -109,13 +109,13 @@ int main() {
     ctx.appendDialectRegistry(registry_);
     mlir::mhlo::registerAllMhloDialects(registry_);
     mlir::stablehlo::registerAllDialects(registry_);
-    ctx.insertenzyme::EnzymeDialect();
 
     auto module_op_ = xla::ConvertHloToStablehlo(ctx, &proto).value().release();
 
     llvm::InitializeNativeTarget();
     llvm::InitializeNativeTargetAsmPrinter();
 
+    registry_.insert<enzyme::EnzymeDialect>();
     registry_.insert<mlir::stablehlo::check::CheckDialect>();
     prepareRegistry(registry_);