nit update

pytorch · Jul 18, 2024 · eeded81 · eeded81
1 parent f0d5541
commit eeded81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/csrc/runtime/pjrt_registry.h b/torch_xla/csrc/runtime/pjrt_registry.h
@@ -22,7 +22,7 @@ void RegisterPjRtPlugin(std::string name,
                         std::shared_ptr<const PjRtPlugin> plugin);
 
 std::tuple<std::unique_ptr<xla::PjRtClient>, std::unique_ptr<XlaCoordinator>>
-InitializePjRt(const std::string& device_type, bool use_aot = false);
+InitializePjRt(const std::string& device_type);
 
 }  // namespace runtime
 }  // namespace torch_xla