From 069029c49977ba9f67d64f5dcf352c7bb6e0bd1c Mon Sep 17 00:00:00 2001
From: Ted Hong <tedhong@google.com>
Date: Tue, 16 Jul 2024 20:23:53 -0700
Subject: [PATCH] Added package() function to proc_base_jit_wrapper.

PiperOrigin-RevId: 653072217
---
 xls/jit/proc_base_jit_wrapper.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/xls/jit/proc_base_jit_wrapper.h b/xls/jit/proc_base_jit_wrapper.h
index 3b02ef0e67..46b4f6be07 100644
--- a/xls/jit/proc_base_jit_wrapper.h
+++ b/xls/jit/proc_base_jit_wrapper.h
@@ -83,6 +83,7 @@ class BaseProcJitWrapper {
         new RealType(std::move(package), proc, std::move(aot), runtime));
   }
 
+  Package* package() const { return package_.get(); }
   ProcRuntime* runtime() const { return runtime_.get(); }
 
   // Reset the state of all of the procs to their initial state.