Add IndexManipulationBenchmarks

Author: lkdvos

benchmark/TensorKitBenchmarks/indexmanipulations/IndexManipulationBenchmarks.jl

@@ -0,0 +1,55 @@
+module IndexManipulationBenchmarks
+
+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"))
+
+# permute!
+# --------
+function init_permute_tensors(T, W, p)
+    C = randn(T, permute(W, p))
+    A = randn(T, W)
+    return C, A
+end
+function benchmark_permute!(benchgroup, params::Dict)
+    haskey(benchgroup, "permute") || addgroup!(benchgroup, "permute")
+    bench = benchgroup["permute"]
+    for kwargs in expand_kwargs(params)
+        benchmark_permute!(bench; kwargs...)
+    end
+    return nothing
+end
+function benchmark_permute!(bench; sigmas=nothing, T="Float64", I="Trivial", dims, p)
+    T_ = parse_type(T)
+    I_ = parse_type(I)
+
+    p_ = (Tuple(p[1]), Tuple(p[2]))
+    Vs = generate_space.(I_, dims, sigmas)
+
+    codomain = mapreduce(Base.Fix1(getindex, Vs), ⊗, p_[1]; init=one(eltype(Vs)))
+    domain = mapreduce(Base.Fix1(getindex, Vs), ⊗, p_[2]; init=one(eltype(Vs)))
+    init() = init_permute_tensors(T_, codomain ← domain, p_)
+
+    bench[T, I, dims, sigmas, p] = @benchmarkable permute!(C, A, $p_) setup = ((C, A) = $init())
+    return nothing
+end
+
+if haskey(all_parameters, "permute")
+    g = addgroup!(SUITE, "permute")
+    for params in all_parameters["permute"]
+        benchmark_permute!(g, params)
+    end
+end
+
+# transpose!
+# ----------
+
+# TODO
+
+end

benchmark/TensorKitBenchmarks/indexmanipulations/benchparams.toml

@@ -0,0 +1,12 @@
+[[permute]]
+T = ["Float64"]
+I = "Trivial"
+p = [[[2, 1], []]]
+dims = [[7264, 7264], [43408, 1216]]
+
+[[permute]]
+T = ["Float64"]
+I = "Z2Irrep"
+p = [[[2, 1], []]]
+dims = [[7264, 7264], [43408, 1216]]
+sigmas = [[0.5, 0.5]]