From b2a02084bb520e4eda46f60f73e6c4cd1c428fcb Mon Sep 17 00:00:00 2001 From: Alan Richardson Date: Tue, 26 Sep 2023 17:06:15 +0100 Subject: [PATCH] Remove unused JIT test Deepwave no longer supports torch.jit.script. This commit removes the test that applied torch.jit.script to the scalar propagator, and which was previously commented-out. --- tests/test_scalar.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/tests/test_scalar.py b/tests/test_scalar.py index f2ccd19..c36bef1 100644 --- a/tests/test_scalar.py +++ b/tests/test_scalar.py @@ -600,17 +600,6 @@ def test_gradcheck_only_wavefield_0_2d(): ) -#def test_jit(): -# """Test that the propagator can be JIT compiled""" -# torch.jit.script(Scalar(torch.ones(1, 1), 5.0))( -# 0.001, source_amplitudes=torch.ones(1, 1, 1), -# source_locations=torch.zeros(1, 1, 2) -# ) -# torch.jit.script(scalar)(torch.ones(1, 1), 5.0, 0.001, -# source_amplitudes=torch.ones(1, 1, 1), -# source_locations=torch.zeros(1, 1, 2).long()) - - def direct_2d_approx(x, x_s, dx, dt, c, f): """Use an approximation of the 2D Green's function to calculate the wavefield at a given location due to the given source.