From 530b7a7c1b149460eb4300f9eb47a65f4d0a9257 Mon Sep 17 00:00:00 2001 From: wardvermeulen Date: Thu, 12 Oct 2023 22:36:59 +0200 Subject: [PATCH] tropical FPU fix --- src/operator.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/operator.jl b/src/operator.jl index d27a969f..73878e4a 100644 --- a/src/operator.jl +++ b/src/operator.jl @@ -123,8 +123,8 @@ function operator_fma(::Type{FPUOp{M, N, K, mb, nb, kb, CT, AT}}, a::CT, b::CT, return fma(a, b, c) end -abstract type TropicalFPUOp{M, N, K, CT, AT} <: GeneralFPUOp{M, N, K, 4, 8, 1, CT, AT} end -function operator_fma(::Type{TropicalFPUOp{M, N, K, CT, AT}}, a::CT, b::CT, c::AT) where {M, N, K, CT, AT} +abstract type TropicalFPUOp{M, N, K, mb, nb, kb, CT, AT} <: GeneralFPUOp{M, N, K, mb, nb, kb, CT, AT} end +function operator_fma(::Type{TropicalFPUOp{M, N, K, mb, nb, kb, CT, AT}}, a::CT, b::CT, c::AT) where {M, N, K, mb, nb, kb, CT, AT} return max(a + b, c) end