Skip to content

Commit 19921cb

Browse files
committed
[TEST] Update CG solver test to disable GPU during computation
1 parent a410626 commit 19921cb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/test_modules/test_cg/test_cg_solver.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,18 +11,19 @@
1111

1212
pytest.mark.skip("Run explicitly")
1313

14+
1415
def test_solve_with_cg():
1516
model = gp.generate_example_model(ExampleModel.GREENSTONE, compute_model=False)
1617
print(model.structural_frame)
1718

1819
WeightCache.clear_cache()
1920
BackendTensor.PYKEOPS = False
20-
21+
2122
sol = gp.compute_model(
2223
gempy_model=model,
2324
engine_config=gp.data.GemPyEngineConfig(
2425
backend=gp.data.AvailableBackends.PYTORCH,
25-
use_gpu=True,
26+
use_gpu=False,
2627
dtype='float64'
2728
)
2829
)

0 commit comments

Comments
 (0)