Split out BLAS Level 3 GEMM tests (#2610)

Co-authored-by: Tim Besard <[email protected]>

Author: kshyatt

lib/cublas/linalg.jl

@@ -203,19 +203,19 @@ LinearAlgebra.generic_trimatdiv!(C::StridedCuVector{T}, uploc, isunitc, tfun::Fu
     trsv!(uploc, tfun === identity ? 'N' : tfun === transpose ? 'T' : 'C', isunitc, A, C === B ? C : copyto!(C, B))
 
 # work around upstream breakage from JuliaLang/julia#55547
-@static if VERSION == v"1.11.2"
-const CuUpperOrUnitUpperTriangular = LinearAlgebra.UpperOrUnitUpperTriangular{
-    <:Any,<:Union{<:CuArray, Adjoint{<:Any, <:CuArray}, Transpose{<:Any, <:CuArray}}}
-const CuLowerOrUnitLowerTriangular = LinearAlgebra.LowerOrUnitLowerTriangular{
-    <:Any,<:Union{<:CuArray, Adjoint{<:Any, <:CuArray}, Transpose{<:Any, <:CuArray}}}
-
-LinearAlgebra.istriu(::CuUpperOrUnitUpperTriangular) = true
-LinearAlgebra.istril(::CuUpperOrUnitUpperTriangular) = false
-LinearAlgebra.istriu(::CuLowerOrUnitLowerTriangular) = false
-LinearAlgebra.istril(::CuLowerOrUnitLowerTriangular) = true
+@static if VERSION >= v"1.11.2"
+    const CuUpperOrUnitUpperTriangular = LinearAlgebra.UpperOrUnitUpperTriangular{
+        <:Any,<:Union{<:CuArray, Adjoint{<:Any, <:CuArray}, Transpose{<:Any, <:CuArray}}}
+    const CuLowerOrUnitLowerTriangular = LinearAlgebra.LowerOrUnitLowerTriangular{
+        <:Any,<:Union{<:CuArray, Adjoint{<:Any, <:CuArray}, Transpose{<:Any, <:CuArray}}}
+    LinearAlgebra.istriu(::CuUpperOrUnitUpperTriangular) = true
+    LinearAlgebra.istril(::CuUpperOrUnitUpperTriangular) = false
+    LinearAlgebra.istriu(::CuLowerOrUnitLowerTriangular) = false
+    LinearAlgebra.istril(::CuLowerOrUnitLowerTriangular) = true
 end
 
 
+
 #
 # BLAS 3
 #

lib/cusparse/linalg.jl

@@ -238,6 +238,18 @@ function LinearAlgebra.dot(y::CuVector{T}, A::CuSparseMatrixCSR{T}, x::CuVector{
     return sum(result)
 end
 
+# work around upstream breakage from JuliaLang/julia#55547
+@static if VERSION >= v"1.11.2"
+    const CuSparseUpperOrUnitUpperTriangular = LinearAlgebra.UpperOrUnitUpperTriangular{
+        <:Any,<:Union{<:AbstractCuSparseMatrix, Adjoint{<:Any, <:AbstractCuSparseMatrix}, Transpose{<:Any, <:AbstractCuSparseMatrix}}}
+    const CuSparseLowerOrUnitLowerTriangular = LinearAlgebra.LowerOrUnitLowerTriangular{
+        <:Any,<:Union{<:AbstractCuSparseMatrix, Adjoint{<:Any, <:AbstractCuSparseMatrix}, Transpose{<:Any, <:AbstractCuSparseMatrix}}}
+    LinearAlgebra.istriu(::CuSparseUpperOrUnitUpperTriangular) = true
+    LinearAlgebra.istril(::CuSparseUpperOrUnitUpperTriangular) = false
+    LinearAlgebra.istriu(::CuSparseLowerOrUnitLowerTriangular) = false
+    LinearAlgebra.istril(::CuSparseLowerOrUnitLowerTriangular) = true
+end
+
 for SparseMatrixType in [:CuSparseMatrixCSC, :CuSparseMatrixCSR]
     @eval begin
         LinearAlgebra.triu(A::$SparseMatrixType{T}, k::Integer) where {T} =

test/core/initialization.jl

@@ -185,7 +185,9 @@ end
 
 ## allocations
 
-@test @allocated(current_context()) == 0
-@test @allocated(context()) == 0
-@test @allocated(stream()) == 0
-@test @allocated(device()) == 0
+let broken = VERSION == v"1.11.3" && Base.JLOptions().code_coverage != 0
+    @test @allocated(current_context()) == 0 broken=broken
+    @test @allocated(context()) == 0 broken=broken
+    @test @allocated(stream()) == 0 broken=broken
+    @test @allocated(device()) == 0 broken=broken
+end