Skip to content

Commit 6bad5cd

Browse files
Rabab53jpsamaroo
andcommitted
DArray: Implement in-place matrix-matrix multiply
Co-authored-by: Julian P Samaroo <[email protected]>
1 parent 969155b commit 6bad5cd

File tree

4 files changed

+495
-2
lines changed

4 files changed

+495
-2
lines changed

src/Dagger.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ include("array/matrix.jl")
7272
include("array/sparse_partition.jl")
7373
include("array/sort.jl")
7474
include("array/linalg.jl")
75+
include("array/mul.jl")
7576
include("array/cholesky.jl")
7677

7778
# Visualization

src/array/matrix.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,9 @@ end
3535
size(x::MatMul) = mul_size(x.a, x.b)
3636
MatMul(a,b) =
3737
MatMul{promote_type(eltype(a), eltype(b)), length(mul_size(a,b))}(a,b)
38-
(*)(a::ArrayOp, b::ArrayOp) = _to_darray(MatMul(a,b))
38+
3939
# Bonus method for matrix-vector multiplication
4040
(*)(a::ArrayOp, b::Vector) = _to_darray(MatMul(a,PromotePartition(b)))
41-
(*)(a::AbstractArray, b::ArrayOp) = _to_darray(MatMul(PromotePartition(a), b))
4241

4342
function (*)(a::ArrayDomain{2}, b::ArrayDomain{2})
4443
if size(a, 2) != size(b, 1)

0 commit comments

Comments
 (0)