From af17178e09d728d4e6793f16ef01b58b25895ad6 Mon Sep 17 00:00:00 2001
From: Elliot Saba <staticfloat@gmail.com>
Date: Mon, 10 Jun 2024 14:45:18 -0700
Subject: [PATCH] Transition to `LinearAlgebra.BLAS.libblastrampoline` instead
 of `Base.libblas_name`

This change is in support of https://github.com/JuliaLang/julia/pull/50699
---
 src/NNlib.jl | 2 --
 src/gemm.jl  | 6 ++++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/NNlib.jl b/src/NNlib.jl
index 7bf7bd172..644a57f3d 100644
--- 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
index 3552f3124..9cbe50195 100644
--- 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!()