From 31bf9fe9ce64aae04969f7843073d1fda075a23a Mon Sep 17 00:00:00 2001 From: Jofre Date: Fri, 3 May 2024 12:38:50 +0200 Subject: [PATCH] Format code --- ext/TenetDaggerExt.jl | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ext/TenetDaggerExt.jl b/ext/TenetDaggerExt.jl index 8de24c5d..e9f93f72 100644 --- a/ext/TenetDaggerExt.jl +++ b/ext/TenetDaggerExt.jl @@ -41,10 +41,11 @@ selectdims(a, proj::Pair...) = function Dagger.stage(ctx::Context, op::Contract{T,N}) where {T,N} domain = isempty(size(op)) ? Dagger.ArrayDomain((1:1, 1:1)) : Dagger.ArrayDomain([1:l for l in size(op)]) - partitioning = isempty(size(op)) ? Dagger.Blocks(1,1) : Dagger.Blocks(op) + partitioning = isempty(size(op)) ? Dagger.Blocks(1, 1) : Dagger.Blocks(op) subdoms_start = tuple([1 for _ in 1:length(partitioning.blocksize)]...) - subdoms_cumlength = tuple([collect(partitioning.blocksize[i]:domain.indexes[i].stop) for i in 1:length(partitioning.blocksize)]...) + subdoms_cumlength = + tuple([collect(partitioning.blocksize[i]:domain.indexes[i].stop) for i in 1:length(partitioning.blocksize)]...) subdomains = Dagger.DomainBlocks{length(partitioning.blocksize)}(subdoms_start, subdoms_cumlength) contractor = @@ -66,7 +67,7 @@ function Dagger.stage(ctx::Context, op::Contract{T,N}) where {T,N} chunks_a = reduce(zip(outer_perm_a, outer_indices_a); init = Dagger.chunks(op.a)) do acc, (d, i) selectdim(acc, d, i:i) end - else + else chunks_a = Dagger.chunks(op.a) end chunks_a = permutedims(chunks_a, vcat(outer_perm_a, inner_perm_a))