Skip to content

Commit

Permalink
simplify MulAdd type in matmul
Browse files Browse the repository at this point in the history
  • Loading branch information
jishnub committed Aug 25, 2023
1 parent cf56520 commit 5fd6999
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions src/generic/matmul.jl
Original file line number Diff line number Diff line change
Expand Up @@ -245,8 +245,7 @@ end

### BandedMatrix * dense matrix

function materialize!(M::MulAdd{<:BandedColumns, <:AbstractColumnMajor, <:AbstractColumnMajor,
T, <:AbstractMatrix, <:AbstractMatrix, <:AbstractMatrix}) where {T}
function materialize!(M::MatMulMatAdd{<:BandedColumns, <:AbstractColumnMajor, <:AbstractColumnMajor})
α, β, A, B, C = M.α, M.β, M.A, M.B, M.C

mA, nA = size(A)
Expand All @@ -265,8 +264,7 @@ function materialize!(M::MulAdd{<:BandedColumns, <:AbstractColumnMajor, <:Abstra
return C
end

function materialize!(M::MulAdd{<:AbstractColumnMajor, <:BandedColumns, <:AbstractColumnMajor,
T, <:AbstractMatrix, <:AbstractMatrix, <:AbstractMatrix}) where {T}
function materialize!(M::MatMulMatAdd{<:AbstractColumnMajor, <:BandedColumns, <:AbstractColumnMajor})
α, β, A, B, C = M.α, M.β, M.A, M.B, M.C
mA, nA = size(A)
mB, nB = size(B)
Expand Down

0 comments on commit 5fd6999

Please sign in to comment.