diff options
| -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]]; |