broadcast_allocation_failure_mwe.jl

using Test, SparseArrays, LinearAlgebra, Random

N, M, p = 10, 12, 0.3;
Q, X, Y, Z = sprand(N, M, p), sprand(Float32, N, M, p), sprand(N, M, p), sprand(N, M, p);

f(x, y, z) = x + y + z + 1;

broadcast!(+, Q, X, Y, Z); # warmup for @allocated
broadcast!(*, Q, X, Y, Z); # warmup for @allocated
broadcast!(f, Q, X, Y, Z); # warmup for @allocated

@test (@allocated broadcast!(+, Q, X, Y, Z)) == 0
@test (@allocated broadcast!(*, Q, X, Y, Z)) == 0
@test (@allocated broadcast!(f, Q, X, Y, Z)) == 0