Updating disctonv layer norm impl

LLNL · May 24, 2023 · 13aff25 · 13aff25
1 parent 8c3d5ca
commit 13aff25
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
@@ -179,6 +179,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];