Transition to LinearAlgebra.BLAS.libblastrampoline instead of `Base…

….libblas_name` This change is in support of JuliaLang/julia#50699
staticfloat · Jun 10, 2024 · af17178 · af17178
1 parent 7369244
commit af17178
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/NNlib.jl b/src/NNlib.jl
@@ -18,8 +18,6 @@ using Requires
 using Statistics
 using Statistics: mean
 
-const libblas = Base.libblas_name
-
 const Numeric = Union{AbstractArray{<:T}, T} where {T<:Number}
 
 # Include APIs

diff --git a/src/gemm.jl b/src/gemm.jl
@@ -3,6 +3,12 @@
 
 using LinearAlgebra.BLAS: get_num_threads, set_num_threads
 
+if isdefined(LinearAlgebra.BLAS, :libblastrampoline)
+    const libblas = LinearAlgebra.BLAS.libblastrampoline
+else
+    const libblas = Base.libblas_name
+end
+
 """
     gemm!()