Updating disctonv layer norm impl

LLNL · Jun 25, 2024 · 9e44581 · 9e44581
1 parent 2bc7d22
commit 9e44581
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/layers/regularizers/distconv/distconv_layer_norm.cu b/src/layers/regularizers/distconv/distconv_layer_norm.cu
@@ -181,6 +181,7 @@ void LayerNormalization<Backend, DataType>::calculate_backward_stats(
   const DCTensor<Allocator>& statistics,
   DCTensor<Allocator>& statistics_grad)
 {
+
   const auto& input_dims = input.get_local_shape();
   const auto& statistics_dims = statistics.get_local_shape();
   const auto local_num_samples = input_dims[3];