diff --git a/src/NNlib.jl b/src/NNlib.jl index 7bf7bd17..644a57f3 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 3552f312..9cbe5019 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!()