Skip running Tessellate backpropagation if the matrix is empty (#2465)

LLNL · Aug 12, 2024 · 6adcc45 · 6adcc45
1 parent b41221c
commit 6adcc45
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/layers/transform/tessellate_cutensor_impl.hpp b/src/layers/transform/tessellate_cutensor_impl.hpp
@@ -49,6 +49,8 @@ void tessellate_layer<TensorDataType, T_layout, Dev>::bp_compute_cutensor(
   LocalMat const& input =
     static_cast<LocalMat const&>(output_grad.LockedMatrix());
   LocalMat& output = static_cast<LocalMat&>(input_grad);
+  if (input.IsEmpty() || output.IsEmpty())
+    return;
 
   // Make dimensions and modes
   int dims = static_cast<int>(output_dims.size());