aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBarry Wardell <barry.wardell@gmail.com>2011-05-12 19:28:05 +0200
committerBarry Wardell <barry.wardell@gmail.com>2011-05-12 19:28:05 +0200
commit7e3abe9d51900eb50c733b8ce60b9565789d7d63 (patch)
treed408bcd746df9b033dd106b64ee6c53fd7d58829
parentf5ffe80093f6d7548846cb2529cb3e0258bb395c (diff)
Re-add missing Checkpoint tag to tensors.
-rw-r--r--Tools/CodeGen/TensorToolsKranc.m4
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]];