Merge pull request #13 from theabhirath/master

Use Base.libblas_name for compatibility with Julia master
chengchingwen · Mar 19, 2022 · 58c8496 · 58c8496
2 parents 6c1de87 + c7c4a54
commit 58c8496
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/NeuralAttentionlib.jl b/src/NeuralAttentionlib.jl
@@ -9,6 +9,8 @@ import Adapt: adapt_structure, adapt
 
 import Base
 
+const libblas = Base.libblas_name
+
 export attention, generic_attention
 
 include("./utils.jl")

diff --git a/src/matmul/gemm.jl b/src/matmul/gemm.jl
@@ -13,7 +13,7 @@ for (gemm, elty) in NNlib.gemm_datatype_mappings
             ptrB::Ptr{$elty}, ldb::Int, beta::($elty),
             ptrC::Ptr{$elty}, ldc::Int)
 
-            ccall((BLAS.@blasfunc($gemm), BLAS.libblas), Nothing,
+            ccall((BLAS.@blasfunc($gemm), libblas), Nothing,
                   (Ref{UInt8}, Ref{UInt8}, Ref{BLAS.BlasInt}, Ref{BLAS.BlasInt}, Ref{BLAS.BlasInt},
                    Ref{$elty}, Ptr{$elty}, Ref{BLAS.BlasInt},
                    Ptr{$elty}, Ref{BLAS.BlasInt}, Ref{$elty},