Skip to content

Commit

Permalink
Remove unused JIT test
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
ar4 committed Sep 26, 2023
1 parent 8b4a429 commit b2a0208
Showing 1 changed file with 0 additions and 11 deletions.
11 changes: 0 additions & 11 deletions tests/test_scalar.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit b2a0208

Please sign in to comment.