/
vector_speedtest.jl
42 lines (33 loc) · 1.15 KB
/
vector_speedtest.jl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
using LinearAlgebra
using JuMP
function test(n::Int)
model = Model()
a = rand(n, n)
@variable(model, x[1:n, 1:n])
b = rand(n, n, n)
@variable(model, y[1:n, 1:n, 1:n])
c = rand(n)
@variable(model, z[1:n])
#initialize
@constraint(model, sum(c[i] * z[i] for i=1:n) <= 0)
#Vector
elapsed = @elapsed @constraint(model, sum(c[i] * z[i] for i=1:n) <= 1)
println("Vector with sum(): $(elapsed)")
elapsed = @elapsed @constraint(model, dot(c, z) <= 1)
println("Vector with vecdot() : $(elapsed)")
#2D Matrix
elapsed = @elapsed @constraint(model, sum(a[i, j] * x[i, j] for i=1:n, j=1:n) <= 1)
println("2D Matrix with sum(): $(elapsed)")
elapsed = @elapsed @constraint(model, dot(a, x) <= 1)
println("2D Matrix with bigvecdot(): $(elapsed)")
#3D Matrix
elapsed = @elapsed @constraint(model, sum(b[i, j, k] * y[i, j, k] for i=1:n, j=1:n, k=1:n) <= 1)
println("3D Matrix with sum(): $(elapsed)")
elapsed = @elapsed @constraint(model, dot(b, y) <= 1)
println("3D Matrix with vecdot(): $(elapsed)")
return 0
end
for n in [10, 50, 100, 200, 300]
println("n = $n")
test(n)
end