Skip to content

Commit e141021

Browse files
committed
Add timing information to debug messages
1 parent 84f04ab commit e141021

File tree

1 file changed

+15
-6
lines changed

1 file changed

+15
-6
lines changed

src/tensors/treetransformers.jl

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ struct AbelianTreeTransformer{T,N} <: TreeTransformer
1414
end
1515

1616
function AbelianTreeTransformer(transform, p, Vdst, Vsrc)
17+
t₀ = Base.time()
1718
permute(Vsrc, p) == Vdst || throw(SpaceMismatch("Incompatible spaces for permuting."))
1819
structure_dst = fusionblockstructure(Vdst)
1920
structure_src = fusionblockstructure(Vsrc)
@@ -37,6 +38,10 @@ function AbelianTreeTransformer(transform, p, Vdst, Vsrc)
3738
# sort by (approximate) weight to facilitate multi-threading strategies
3839
# sort!(transformer)
3940

41+
Δt = Base.time() - t₀
42+
43+
@debug("Treetransformer for $Vsrc to $Vdst via $p", nblocks = L, Δt)
44+
4045
return transformer
4146
end
4247

@@ -48,6 +53,7 @@ struct GenericTreeTransformer{T,N} <: TreeTransformer
4853
end
4954

5055
function GenericTreeTransformer(transform, p, Vdst, Vsrc)
56+
t₀ = Base.time()
5157
permute(Vsrc, p) == Vdst || throw(SpaceMismatch("Incompatible spaces for permuting."))
5258
structure_dst = fusionblockstructure(Vdst)
5359
structure_src = fusionblockstructure(Vsrc)
@@ -92,15 +98,18 @@ function GenericTreeTransformer(transform, p, Vdst, Vsrc)
9298
structure_src.fusiontreestructure[ids_src])
9399
end
94100

95-
@debug("TreeTransformer for $Vsrc to $Vdst via $p",
96-
nblocks = length(data),
97-
sz_median = size(data[end ÷ 2][1], 1),
98-
sz_max = size(data[1][1], 1))
99-
100101
transformer = GenericTreeTransformer{T,N}(data)
101102

102103
# sort by (approximate) weight to facilitate multi-threading strategies
103-
# sort!(transformer)
104+
sort!(transformer)
105+
106+
Δt = Base.time() - t₀
107+
108+
@debug("TreeTransformer for $Vsrc to $Vdst via $p",
109+
nblocks = length(data),
110+
sz_median = size(data[end ÷ 2][1], 1),
111+
sz_max = size(data[1][1], 1),
112+
Δt)
104113

105114
return transformer
106115
end

0 commit comments

Comments
 (0)