Fixed APEX test

Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com>

megatron/model/transformer.py

@@ -21,7 +21,7 @@ import torch
 try:
     from apex.normalization.fused_layer_norm import FusedLayerNorm as LayerNorm
     # Try to use FusedLayerNorm from Apex - this will trigger an error.
-    _ = FusedLayerNorm(8, eps=1e-5)
+    _ = LayerNorm(8, eps=1e-5)
 
 except Exception as e:
     print('WARNING: APEX is not available, using torch.nn.LayerNorm instead of apex.normalization.FusedLayerNorm!')