Add caching improvements and enhance solver stability

Introduced a `clear_cache` method in the `WeightCache` class for better memory management. Enhanced the numerical stability of the conjugate gradient solver with improved initialization and adaptive tolerances. Refactored interpolation logic to handle weights more efficiently, and adjusted benchmarking to test expanded solver configurations.

Author: Leguark

gempy/modules/data_manipulation/_engine_factory.py

@@ -42,11 +42,18 @@ def interpolation_input_from_structural_frame(geo_model: "gempy.data.GeoModel")
         extent_transformed=geo_model.extent_transformed_transformed_by_input
     )
 
+    weights = []
+    if geo_model.solutions is not None:
+        for stack_sol in geo_model.solutions.root_output.outputs_centers:
+            weights.append(stack_sol.weights)
+    
+
     interpolation_input: InterpolationInput = InterpolationInput(
         surface_points=surface_points,
         orientations=orientations,
         grid=grid,
-        unit_values=structural_frame.elements_ids  # TODO: Here we will need to pass densities etc.
+        unit_values=structural_frame.elements_ids,  # TODO: Here we will need to pass densities etc.
+        weights=weights
     )
 
     return interpolation_input

test/test_modules/test_cg/test_cg_solver.py

@@ -1,14 +1,20 @@
 import gempy as gp
 from gempy.core.data.enumerators import ExampleModel
 from gempy.optional_dependencies import require_gempy_viewer
+import gempy_engine.core.backend_tensor as BackendTensor
+
+from gempy_engine.modules.weights_cache.weights_cache_interface import WeightCache
 
 PLOT = True
 
 
-def test_generate_greenstone_model():
+def test_solve_with_cg():
     model = gp.generate_example_model(ExampleModel.GREENSTONE, compute_model=False)
     print(model.structural_frame)
 
+    WeightCache.clear_cache()
+    BackendTensor.PYKEOPS = True
+    
     sol = gp.compute_model(
         gempy_model=model,
         engine_config=gp.data.GemPyEngineConfig(
@@ -21,3 +27,36 @@ def test_generate_greenstone_model():
     if PLOT:
         gpv = require_gempy_viewer()
         gpv.plot_3d(model, image=True)
+
+
+def test_save_weights():
+    model = gp.generate_example_model(ExampleModel.GREENSTONE, compute_model=False)
+    print(model.structural_frame)
+
+    sol = gp.compute_model(
+        gempy_model=model,
+        engine_config=gp.data.GemPyEngineConfig(
+            backend=gp.data.AvailableBackends.PYTORCH,
+            use_gpu=False,
+            dtype='float32'
+        )
+    )
+    weights1 = sol.octrees_output[0].outputs_centers[0].weights
+    weights2 = sol.octrees_output[0].outputs_centers[1].weights
+    weights3 = sol.octrees_output[0].outputs_centers[2].weights
+
+    WeightCache.clear_cache()
+    BackendTensor.PYKEOPS = True
+
+    sol = gp.compute_model(
+        gempy_model=model,
+        engine_config=gp.data.GemPyEngineConfig(
+            backend=gp.data.AvailableBackends.PYTORCH,
+            use_gpu=False,
+            dtype='float32'
+        )
+    )
+
+    if PLOT:
+        gpv = require_gempy_viewer()
+        gpv.plot_3d(model, image=True)