diff --git a/diffsl/src/execution/compiler.rs b/diffsl/src/execution/compiler.rs
index 918cc31..f7a2c5f 100644
--- a/diffsl/src/execution/compiler.rs
+++ b/diffsl/src/execution/compiler.rs
@@ -202,6 +202,10 @@ impl<M: CodegenModule, T: Scalar> Compiler<M, T> {
         &self._module
     }
 
+    pub fn to_module(self) -> M {
+        self._module
+    }
+
     pub fn get_tensors(&self) -> Vec<String> {
         self.jit_get_tensor_functions
             .data_map