Skip to content

Commit

Permalink
Update CuTropicalGEMM.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
ArrogantGao authored Oct 17, 2023
1 parent c6b0e86 commit a2678fc
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/CuTropicalGEMM.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ export matmul!

function __init__()
@assert CUDA.driver_version() >= v"11.4" "Error: CUDA.driver_version < v11.4"
@assert CUDA.driver_version() <= v"12.1" "Error: CUDA.driver_version > v12.1"
@assert CUDA.driver_version() <= v"12.2" "Error: CUDA.driver_version > v12.2"
return nothing
end

Expand All @@ -19,4 +19,4 @@ const CTranspose{T} = Transpose{T, <:CuVecOrMat{T}}

include("tropical_gemms.jl")

end
end

0 comments on commit a2678fc

Please sign in to comment.