Skip to content

Commit

Permalink
Update compile_with_xla.cc (#147)
Browse files Browse the repository at this point in the history
Stop setting registry.  It's no longer used within XLA.
  • Loading branch information
klucke authored Oct 22, 2024
1 parent 37ab04c commit 10155ba
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/enzyme_ad/jax/compile_with_xla.cc
Original file line number Diff line number Diff line change
Expand Up @@ -419,7 +419,6 @@ compile_mhlo_to_llvm_with_xla(llvm::StringRef mhlo_text, std::string &output,
xla::Compiler::CompileOptions opts = {
build_options.device_allocator(), build_options.compile_thread_pool(),
build_options.layout_canonicalization_callback()};
opts.registry = &registry;
auto executable =
BuildExecutable(local_client->local_service(), xla_computation.proto(),
std::move(module_config_or_error.value()),
Expand Down

0 comments on commit 10155ba

Please sign in to comment.