remove blas thread that always segfault

slimgroup · Mar 18, 2024 · 9f46a00 · 9f46a00
1 parent 7caaf90
commit 9f46a00
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 12 deletions.
diff --git a/src/JUDI.jl b/src/JUDI.jl
@@ -210,8 +210,6 @@ function __init__()
         end
     end
 
-    # BLAS num threads for dense LA such as sinc interpolation
-    BLAS.set_num_threads(Threads.nthreads())
 end
 
 end
diff --git a/src/pysource/models.py b/src/pysource/models.py
@@ -427,7 +427,7 @@ def _cfl_coeff(self):
             c_fd = sum(np.abs(coeffs[-1][-1])) / 2
             return .9 * np.sqrt(self.dim) / self.dim / c_fd
         a1 = 4  # 2nd order in time
-        so = max(self._space_order, 4)
+        so = max(self._space_order // 2, 4)
         coeffs = fd_w(2, range(-so, so), 0)[-1][-1]
         return .9 * np.sqrt(a1/float(self.grid.dim * sum(np.abs(coeffs))))
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -29,7 +29,6 @@ parsed_args = parse_commandline()
 const nlayer = parsed_args["nlayer"]
 const tti = parsed_args["tti"] || contains(GROUP, "TTI")
 const fs = parsed_args["fs"] || contains(GROUP, "FS")
-const nw = (GROUP == "BASICS" || GROUP == "ISO_OP") ? 2 : 0
 const viscoacoustic = parsed_args["viscoacoustic"] || contains(GROUP, "VISCO")
 
 
@@ -72,7 +71,6 @@ issues = ["test_issues.jl"]
 
 # custom
 if endswith(GROUP, ".jl")
-    # VERSION >= v"1.7" && push!(Base.ARGS, "-p 2")
     timeit_include(GROUP)
 end
 

diff --git a/test/test_adjoint.jl b/test/test_adjoint.jl
@@ -3,7 +3,7 @@
 # Date: May 2020
 #
 
-using Distributed
+nw = 2
 
 # # Set parallel if specified
 if nw > 1 && nworkers() < nw
@@ -60,9 +60,9 @@ test_adjoint(adj::Bool, last::Bool) = (adj || last) ? (@test adj) : (@test_skip
 # Modeling operators
 @testset "Adjoint test with $(nlayer) layers and tti $(tti) and viscoacoustic $(viscoacoustic) and freesurface $(fs)" begin
     @timeit TIMEROUTPUT "Adjoint" begin
-        opt = Options(sum_padding=true, dt_comp=dt, free_surface=parsed_args["fs"], f0=f0)
+        opt = Options(sum_padding=true, dt_comp=dt, free_surface=fs, f0=f0)
         F = judiModeling(model0, srcGeometry, recGeometry; options=opt)
-
+        @show q.nsrc
         # Nonlinear modeling
         y = F*q
 
@@ -84,7 +84,7 @@ end
 # Extended source modeling
 @testset "Extended source adjoint test with $(nlayer) layers and tti $(tti) and viscoacoustic $(viscoacoustic) and freesurface $(fs)" begin
     @timeit TIMEROUTPUT "Extended source adjoint" begin
-        opt = Options(sum_padding=true, dt_comp=dt, free_surface=parsed_args["fs"], f0=f0)
+        opt = Options(sum_padding=true, dt_comp=dt, free_surface=fs, f0=f0)
         F = judiModeling(model0, srcGeometry, recGeometry; options=opt)
         Pr = judiProjection(recGeometry)
         Fw = judiModeling(model0; options=opt)

diff --git a/test/test_linearity.jl b/test/test_linearity.jl
@@ -138,7 +138,7 @@ end
 
 
 ####################### Extended source operators ##########################################
-if parsed_args["tti"]
+if tti
     ftol = 5f-4
 end
 

diff --git a/test/test_modeling.jl b/test/test_modeling.jl
@@ -6,13 +6,14 @@
 # Mathias Louboutin, [email protected]
 # Updated July 2020
 
+nw = 2
+
 # Set parallel if specified
 if nw > 1 && nworkers() < nw
    addprocs(nw-nworkers() + 1; exeflags=["--code-coverage=user", "--inline=no", "--check-bounds=yes"])
 end
 
-@everywhere using JOLI
-@everywhere using JUDI, LinearAlgebra, Test, Distributed
+@everywhere using JOLI, JUDI, LinearAlgebra, Test, Distributed
 
 ### Model
 model, model0, dm = setup_model(tti, viscoacoustic, nlayer; n=(101, 101), d=(10., 10.))