From 13aff2582dbf03138d9f2b20331d20ee8fad829f Mon Sep 17 00:00:00 2001 From: Shehtab Zaman Date: Tue, 23 May 2023 19:56:01 -0700 Subject: [PATCH] Updating disctonv layer norm impl --- src/layers/regularizers/distconv/distconv_layer_norm.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/src/layers/regularizers/distconv/distconv_layer_norm.cu b/src/layers/regularizers/distconv/distconv_layer_norm.cu index 1821cd5b161..9072f739708 100644 --- a/src/layers/regularizers/distconv/distconv_layer_norm.cu +++ b/src/layers/regularizers/distconv/distconv_layer_norm.cu @@ -179,6 +179,7 @@ void LayerNormalization::calculate_backward_stats( const DCTensor& statistics, DCTensor& 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];