Skip to content

Commit

Permalink
Finished reverting Rounding in Config
Browse files Browse the repository at this point in the history
  • Loading branch information
orkolorko committed Dec 20, 2024
1 parent 4de16b1 commit 48c36d0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/device/intrinsics/wmma.jl
Original file line number Diff line number Diff line change
Expand Up @@ -756,7 +756,7 @@ mma
@generated function mma(a::Fragment{M, N, K, A_SZ, A_T, A_L, MatrixA},
b::Fragment{M, N, K, B_SZ, B_T, B_L, MatrixB},
c::Fragment{M, N, K, C_SZ, C_T, Unspecified, Accumulator},
config::Type{Config{M, N, K, D_T}}; rounding = RoundNearest) where {M, N, K, A_SZ, A_T, A_L, B_SZ, B_T, B_L, C_SZ, C_T, D_T}
config::Type{Config{M, N, K, D_T}}) where {M, N, K, A_SZ, A_T, A_L, B_SZ, B_T, B_L, C_SZ, C_T, D_T}

a_layout = get_hl_layout(A_L)
b_layout = get_hl_layout(B_L)
Expand Down

0 comments on commit 48c36d0

Please sign in to comment.