Add support for DefineDerivative with two arguments to TensorTools

ianhinder · Mar 11, 2014 · d83c700 · d83c700
1 parent 8b42593
commit d83c700
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Tools/CodeGen/TensorTools.m b/Tools/CodeGen/TensorTools.m
@@ -292,6 +292,8 @@
     TensorAttributes[T] = {TensorWeight -> 0, Symmetries -> {}};
     T];
 
+DefineDerivative[pd_, nd_] := (pd = nd; DefineDerivative[nd]);
+
 DefineDerivative[pd_Symbol] :=
   If[!DerivativeOperatorQ[pd], AppendTo[$Derivatives, pd]];