From 51849ab2a3debfc075d87208e12680d9d0e91d18 Mon Sep 17 00:00:00 2001 From: Brian Chen Date: Mon, 24 Oct 2022 22:21:43 -0700 Subject: [PATCH] BLAS threads = 1 --- test/runtests.jl | 3 +++ 1 file changed, 3 insertions(+) diff --git a/test/runtests.jl b/test/runtests.jl index 397131b98..efcd74b72 100644 --- 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