diff options
author | Barry Wardell <barry.wardell@gmail.com> | 2011-05-12 19:28:05 +0200 |
---|---|---|
committer | Barry Wardell <barry.wardell@gmail.com> | 2011-05-12 19:28:05 +0200 |
commit | 7e3abe9d51900eb50c733b8ce60b9565789d7d63 (patch) | |
tree | d408bcd746df9b033dd106b64ee6c53fd7d58829 | |
parent | f5ffe80093f6d7548846cb2529cb3e0258bb395c (diff) |
Re-add missing Checkpoint tag to tensors.
-rw-r--r-- | Tools/CodeGen/TensorToolsKranc.m | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/Tools/CodeGen/TensorToolsKranc.m b/Tools/CodeGen/TensorToolsKranc.m index ebac9e6..1abda8b 100644 --- a/Tools/CodeGen/TensorToolsKranc.m +++ b/Tools/CodeGen/TensorToolsKranc.m @@ -76,6 +76,10 @@ CreateGroupFromTensor[k_, inds_] := reflectionParityString[GetTensorAttribute[k, TensorManualCartesianParities]]]]; If[HasTensorAttribute[k, TensorParity], tags = Append[tags, "tensorparity" -> GetTensorAttribute[k, TensorParity]]]; + + If[HasTensorAttribute[k, Checkpoint], + tags = Append[tags, "checkpoint" -> GetTensorAttribute[k, Checkpoint]]]; + vars = If[nInds == 0, {k}, {Apply[Tensor, {k, Apply[Sequence,inds]}]}]; group = CreateGroup[ToString[k] <> "_group", vars, {Tags -> tags}]; Return[group]]; |