getting there

Author: dschwen

src/tensor_computes/NEML2TensorCompute.C

@@ -8,6 +8,7 @@
 
 #pragma once
 
+#include "Conversion.h"
 #include "NEML2TensorCompute.h"
 #include "NEML2Utils.h"
 
@@ -70,6 +71,7 @@ NEML2TensorCompute::NEML2TensorCompute(const InputParameters & params)
   const auto inputs = getParam<TensorInputBufferName, std::string>("swift_inputs", "neml2_inputs");
   std::map<neml2::LabeledAxisAccessor, TensorInputBufferName> lookup_swift_name;
   const auto model_inputs = _model.consumed_items();
+  mooseInfo("Inputs: ", Moose::stringify(model_inputs));
 
   // current inputs
   for (const auto & [swift_input_name, neml2_input_name] : inputs)
@@ -98,7 +100,7 @@ NEML2TensorCompute::NEML2TensorCompute(const InputParameters & params)
 
   // old state inputs
   for (const auto & neml2_input : model_inputs)
-    if (neml2_input.is_old_state())
+    if (neml2_input.is_old_state() || neml2_input.is_old_force())
     {
       // check if we couple the current state
       auto it = lookup_swift_name.find(neml2_input.current());

src/tensor_computes/TimeTensorCompute.C

@@ -27,5 +27,5 @@ TimeTensorCompute::TimeTensorCompute(const InputParameters & parameters)
 void
 TimeTensorCompute::computeBuffer()
 {
-  _u = torch::tensor({_time}, MooseTensor::floatTensorOptions());
+  _u = torch::tensor({_time}, MooseTensor::floatTensorOptions()).expand(_domain.getShape());
 }

test/tests/neml2/old_state.i

@@ -18,25 +18,29 @@
       buffer = A
       real = 2
     []
+  []
+  [Solve]
+    [time]
+      type = TimeTensorCompute
+      buffer = time
+    []
     [dAdt]
       type = NEML2TensorCompute
       neml2_input_file = neml2_input.i
       neml2_model = rate
-      swift_inputs = 'A B'
-      neml2_inputs = 'forces/A forces/B'
+      swift_inputs = 'A time'
+      neml2_inputs = 'forces/A forces/t'
       neml2_outputs = 'state/dAdt'
       swift_outputs = 'dAdt'
     []
   []
-  [Solve]
-
-  []
 []
 
 [Postprocessors]
   [dAdt]
     type = TensorAveragePostprocessor
     buffer = dAdt
+    execute_on = 'TIMESTEP_END'
   []
 []