Skip to content

Commit 29c6d35

Browse files
committed
Update
1 parent 9d79573 commit 29c6d35

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

test/dispatch.jl

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,13 @@ function dispatch_tests(::Type{T}) where {T}
1515
c = one(T)
1616
x = convert.(T, [1, 2, 3])
1717
# Need to allocate 1 BigInt for the result and one for the buffer
18-
alloc_test(() -> MA.fused_map_reduce(MA.add_mul, x, x), 2BIGINT_ALLOC)
19-
alloc_test(() -> MA.fused_map_reduce(MA.add_dot, x, x), 2BIGINT_ALLOC)
18+
nalloc = 2 * (sizeof(Int) + _big_int_alloc())
19+
alloc_test(() -> MA.fused_map_reduce(MA.add_mul, x, x), nalloc)
20+
alloc_test(() -> MA.fused_map_reduce(MA.add_dot, x, x), nalloc)
2021
if T <: MA.AbstractMutable
21-
alloc_test(() -> x'x, 2BIGINT_ALLOC)
22-
alloc_test(() -> transpose(x) * x, 2BIGINT_ALLOC)
23-
alloc_test(() -> LinearAlgebra.dot(x, x), 2BIGINT_ALLOC)
22+
alloc_test(() -> x'x, nalloc)
23+
alloc_test(() -> transpose(x) * x, nalloc)
24+
alloc_test(() -> LinearAlgebra.dot(x, x), nalloc)
2425
end
2526
end
2627

0 commit comments

Comments
 (0)