"examples/vscode:/vscode.git/clone" did not exist on "ff43dba7eaee6d2055d299cf58183b8d19a35daa"
Commit da1c96e9 authored by Deepak Narayanan's avatar Deepak Narayanan
Browse files

Add assertion for now preventing usage of pipeline parallelism with T5 model

parent 2babcaf6
...@@ -331,6 +331,10 @@ class TransformerLanguageModel(MegatronModule): ...@@ -331,6 +331,10 @@ class TransformerLanguageModel(MegatronModule):
# Decoder (usually set to False, True if part of an encoder-decoder # Decoder (usually set to False, True if part of an encoder-decoder
# architecture and in decoder-only stage). # architecture and in decoder-only stage).
if self.add_decoder: if self.add_decoder:
# Temporary assertion until we verify correctness of pipeline parallelism
# implementation of T5.
assert args.pipeline_model_parallel_size == 1, \
'pipeline parallelism is not supported in the presence of decoder'
self.decoder = ParallelTransformer( self.decoder = ParallelTransformer(
self.init_method, self.init_method,
output_layer_init_method, output_layer_init_method,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment