Add LinalgBenchmarks

Author: lkdvos

benchmark/TensorKitBenchmarks/linalg/LinalgBenchmarks.jl

@@ -0,0 +1,81 @@
+module LinalgBenchmarks
+
+include(joinpath(@__DIR__, "..", "utils", "BenchUtils.jl"))
+
+using .BenchUtils
+using BenchmarkTools
+using TensorKit
+using TOML
+
+const SUITE = BenchmarkGroup()
+const all_parameters = TOML.parsefile(joinpath(@__DIR__, "benchparams.toml"))
+
+# mul!
+# ----
+function init_mul_tensors(T, V)
+    A = randn(T, V[1] ← V[2])
+    B = randn(T, V[2] ← V[3])
+    C = randn(T, V[1] ← V[3])
+    return A, B, C
+end
+
+function benchmark_mul!(benchgroup, params::Dict)
+    haskey(benchgroup, "mul") || addgroup!(benchgroup, "mul")
+    bench = benchgroup["mul"]
+    for kwargs in expand_kwargs(params)
+        benchmark_mul!(bench; kwargs...)
+    end
+    return nothing
+end
+
+function benchmark_mul!(bench; sigmas=nothing, T="Float64", I="Trivial", dims)
+    T_ = parse_type(T)
+    I_ = parse_type(I)
+
+    Vs = generate_space.(I_, dims, sigmas)
+    init() = init_mul_tensors(T_, Vs)
+
+    bench[T, I, dims, sigmas] = @benchmarkable mul!(C, A, B) setup = ((A, B, C) = $init())
+
+    return nothing
+end
+
+if haskey(all_parameters, "mul")
+    g = addgroup!(SUITE, "mul")
+    for params in all_parameters["mul"]
+        benchmark_mul!(g, params)
+    end
+end
+
+# svd!
+# ----
+function init_svd_tensor(T, V)
+    A = randn(T, V[1] ← V[2])
+    return A
+end
+
+function benchmark_svd!(benchgroup, params::Dict)
+    haskey(benchgroup, "svd") || addgroup!(benchgroup, "svd")
+    bench = benchgroup["svd"]
+    for kwargs in expand_kwargs(params)
+        benchmark_svd!(bench; kwargs...)
+    end
+    return nothing
+end
+function benchmark_svd!(bench; sigmas=nothing, T="Float64", I="Trivial", dims)
+    T_ = parse_type(T)
+    I_ = parse_type(I)
+    Vs = generate_space.(I_, dims, sigmas)
+    init() = init_svd_tensor(T_, Vs)
+    bench[T, I, dims, sigmas] = @benchmarkable tsvd!(A) setup = (A = $init())
+    return nothing
+end
+
+if haskey(all_parameters, "svd")
+    g = addgroup!(SUITE, "svd")
+    for params in all_parameters["svd"]
+        benchmark_svd!(g, params)
+    end
+end
+
+end

benchmark/TensorKitBenchmarks/linalg/benchparams.toml

@@ -0,0 +1,21 @@
+[[mul]]
+T = ["Float64", "ComplexF64"]
+I = "Trivial"
+dims = [[2, 2, 2], [8, 8, 8], [32, 32, 32], [64, 64, 64], [128, 128, 128]]
+
+[[mul]]
+T = ["Float64", "ComplexF64"]
+I = "Z2Irrep"
+dims = [[2, 2, 2], [8, 8, 8], [32, 32, 32], [64, 64, 64], [128, 128, 128]]
+sigmas = [[0.5, 0.5, 0.5]]
+
+[[svd]]
+T = ["Float64", "ComplexF64"]
+I = "Trivial"
+dims = [[2, 2], [8, 8], [32, 32], [64, 64], [128, 128]]
+
+[[svd]]
+T = ["Float64", "ComplexF64"]
+I = "Z2Irrep"
+dims = [[2, 2], [8, 8], [32, 32], [64, 64], [128, 128]]
+sigmas = [[0.5, 0.5]]