Skip to content

Commit

Permalink
revise the binary dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
ArrogantGao committed Oct 19, 2023
1 parent b4b08d3 commit 918f982
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "CuTropicalGEMM"
uuid = "c2b282c3-c9c2-431d-80f7-a1a0561ebe55"
authors = ["Xuanzhao Gao <[email protected]> and contributors"]
version = "0.1.0"
version = "0.1.1"

[deps]
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
Expand All @@ -11,7 +11,7 @@ TropicalNumbers = "b3a74e9c-7526-4576-a4eb-79c0d4c32334"

[compat]
CUDA = "5"
TropicalGemmC_jll = "0.1"
TropicalGemmC_jll = "0.1.1"
TropicalNumbers = "0.6.2"
julia = "1"

Expand Down
6 changes: 3 additions & 3 deletions src/tropical_gemms.jl
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,9 @@ for (TA, tA) in [(:CuVecOrMat, 'N'), (:CTranspose, 'T')]
for (TB, tB) in [(:CuVecOrMat, 'N'), (:CTranspose, 'T')]
for (TT, CT, funcname, lib) in [
(:Float32, :Cfloat, :FLOAT_plusmul, :lib_PlusMul_FP32), (:Float64, :Cdouble, :DOUBLE_plusmul, :lib_PlusMul_FP64), (:Int32, :Cint, :INT_plusmul, :lib_PlusMul_INT32), (:Int64, :Clong, :LONG_plusmul, :lib_PlusMul_INT64),
(:TropicalAndOr, :Bool, :BOOL_andor, :TropicalAndOr_Bool),
(:TropicalMaxPlusF32, :Cfloat, :FLOAT_maxplus, :TropicalMaxPlus_FP32), (:TropicalMaxPlusF64, :Cdouble, :DOUBLE_maxplus, :TropicalMaxPlus_FP64),
(:TropicalMinPlusF32, :Cfloat, :FLOAT_minplus, :TropicalMinPlus_FP32), (:TropicalMinPlusF64, :Cdouble, :DOUBLE_minplus, :TropicalMinPlus_FP64),
(:TropicalAndOr, :Bool, :BOOL_andor, :lib_TropicalAndOr_Bool),
(:TropicalMaxPlusF32, :Cfloat, :FLOAT_maxplus, :lib_TropicalMaxPlus_FP32), (:TropicalMaxPlusF64, :Cdouble, :DOUBLE_maxplus, :lib_TropicalMaxPlus_FP64),
(:TropicalMinPlusF32, :Cfloat, :FLOAT_minplus, :lib_TropicalMinPlus_FP32), (:TropicalMinPlusF64, :Cdouble, :DOUBLE_minplus, :lib_TropicalMinPlus_FP64),
(:TropicalMaxMulF32, :Cfloat, :FLOAT_maxmul, :lib_TropicalMaxMul_FP32), (:TropicalMaxMulF64, :Cdouble, :DOUBLE_maxmul, :lib_TropicalMaxMul_FP64), (:TropicalMaxMulI32, :Cint, :INT_maxmul, :lib_TropicalMaxMul_INT32), (:TropicalMaxMulI64, :Clong, :LONG_maxmul, :lib_TropicalMaxMul_INT64)
]
@eval function matmul!(C::CuVecOrMat{T}, A::$TA{T}, B::$TB{T}, α::T, β::T) where {T<:$TT}
Expand Down

0 comments on commit 918f982

Please sign in to comment.