BLAS threads = 1

FluxML · Oct 25, 2022 · 51849ab · 51849ab
1 parent c3ee4ab
commit 51849ab
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -7,6 +7,8 @@ using Zygote: gradient
 using StableRNGs
 using CUDA
 
+using LinearAlgebra
+
 const rng = StableRNG(123)
 include("test_utils.jl")
 
@@ -46,6 +48,7 @@ include("test_utils.jl")
     =#
 
     @testset "Convolution" begin
+        BLAS.set_num_threads(1)
         include("conv.jl")
         # include("conv_bias_act.jl")
     end