3outeille's picture
3outeille HF staff
Upload llama-1B/16_GPUS/dp-1_tp-1_pp-16_mbz-32
4507615 verified
raw
history blame
223 kB
========================
START TIME: Tue Jul 2 19:40:59 UTC 2024
python3 version = Python 3.10.14
========================
The token has not been saved to the git credentials helper. Pass `add_to_git_credential=True` in this function directly or `--add-to-git-credential` if using via `huggingface-cli` if you want to set the git credential as well.
Token is valid (permission: write).
Your token has been saved to /admin/home/ferdinand_mom/.cache/huggingface/token
Login successful
Already on 'bench_cluster'
M examples/config_tiny_llama.py
M examples/config_tiny_llama.yaml
M examples/train_tiny_llama.sh
M src/nanotron/models/llama.py
M src/nanotron/trainer.py
Your branch is up to date with 'origin/bench_cluster'.
Job status: RUNNING
W0702 19:41:02.894000 140644320167744 torch/distributed/run.py:757]
W0702 19:41:02.894000 140644320167744 torch/distributed/run.py:757] *****************************************
W0702 19:41:02.894000 140644320167744 torch/distributed/run.py:757] Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed.
W0702 19:41:02.894000 140644320167744 torch/distributed/run.py:757] *****************************************
W0702 19:41:06.424000 140532397197120 torch/distributed/run.py:757]
W0702 19:41:06.424000 140532397197120 torch/distributed/run.py:757] *****************************************
W0702 19:41:06.424000 140532397197120 torch/distributed/run.py:757] Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed.
W0702 19:41:06.424000 140532397197120 torch/distributed/run.py:757] *****************************************
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Config:
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Config(general=GeneralArgs(project='bench_cluster',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: run='%date_%jobid',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: seed=42,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: step=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: consumed_train_samples=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: benchmark_csv_path=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: ignore_sanity_checks=True),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: parallelism=ParallelismArgs(dp=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: pp=16,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tp=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: pp_engine=<nanotron.parallel.pipeline_parallel.engine.OneForwardOneBackwardPipelineEngine object at 0x7f41b0bf4910>,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tp_mode=<TensorParallelLinearMode.REDUCE_SCATTER: 2>,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tp_linear_async_communication=False,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: expert_parallel_size=1),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: model=ModelArgs(model_config=LlamaConfig(bos_token_id=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: eos_token_id=2,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: hidden_act='silu',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: hidden_size=2048,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: initializer_range=0.02,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: intermediate_size=4096,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: is_llama_config=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: max_position_embeddings=4096,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_attention_heads=32,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_hidden_layers=24,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_key_value_heads=32,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: pad_token_id=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: pretraining_tp=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: rms_norm_eps=1e-05,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: rope_scaling=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: rope_theta=10000.0,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tie_word_embeddings=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: use_cache=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: vocab_size=50257),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: init_method=RandomInit(std=0.025),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: dtype=torch.bfloat16,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: make_vocab_size_divisible_by=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: ddp_bucket_cap_mb=25),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tokenizer=TokenizerArgs(tokenizer_name_or_path='openai-community/gpt2',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tokenizer_revision=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tokenizer_max_length=None),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: checkpoints=CheckpointsArgs(checkpoints_path=Path('/dev/null'),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: checkpoint_interval=100000,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: save_initial_state=False,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: resume_checkpoint_path=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: checkpoints_path_is_shared_file_system=False),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: logging=LoggingArgs(log_level='info',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: log_level_replica='info',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: iteration_step_info_interval=1),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tokens=TokensArgs(sequence_length=4096,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: train_steps=20,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: micro_batch_size=32,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: batch_accumulation_per_replica=32,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: val_check_interval=-1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: limit_val_batches=0,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: limit_test_batches=0),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: optimizer=OptimizerArgs(optimizer_factory=AdamWOptimizerArgs(adam_eps=1e-08,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: adam_beta1=0.9,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: adam_beta2=0.95,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: torch_adam_is_fused=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: name='adamW'),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: zero_stage=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: weight_decay=0.01,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: clip_grad=1.0,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: accumulate_grad_in_fp32=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: learning_rate_scheduler=LRSchedulerArgs(learning_rate=0.0001,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: lr_warmup_steps=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: lr_warmup_style='linear',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: lr_decay_style='linear',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: lr_decay_steps=19,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: lr_decay_starting_step=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: min_decay_lr=1e-05)),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: data_stages=[DatasetStageArgs(name='Training Stage',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: start_training_step=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: data=DataArgs(dataset=PretrainDatasetsArgs(hf_dataset_or_datasets='roneneldan/TinyStories',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: hf_dataset_splits='train',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: hf_dataset_config_name=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: dataset_processing_num_proc_per_process=64,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: dataset_overwrite_cache=False,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: text_column_name='text'),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: seed=42,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_loading_workers=32))],
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: profiler=ProfilerArgs(profiler_export_path=Path('/fsx/ferdinandmom/ferdinand-hf/bench_cluster/results/llama-1B/16_GPUS/dp-1_tp-1_pp-16_mbz-32')),
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: lighteval=None)
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Model Config:
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: LlamaConfig(bos_token_id=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: eos_token_id=2,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: hidden_act='silu',
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: hidden_size=2048,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: initializer_range=0.02,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: intermediate_size=4096,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: is_llama_config=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: max_position_embeddings=4096,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_attention_heads=32,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_hidden_layers=24,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: num_key_value_heads=32,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: pad_token_id=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: pretraining_tp=1,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: rms_norm_eps=1e-05,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: rope_scaling=None,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: rope_theta=10000.0,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: tie_word_embeddings=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: use_cache=True,
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: vocab_size=50257)
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Building model..
[default0]:07/02/2024 19:41:28 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Setting PP block ranks...
[default1]:07/02/2024 19:41:45 [INFO|DP=0|PP=9|TP=0|ip-26-0-173-202]: Local number of parameters: 83.9M (160.02MiB)
[default1]:07/02/2024 19:41:45 [INFO|DP=0|PP=9|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default1]:07/02/2024 19:41:45 [INFO|DP=0|PP=9|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default3]:07/02/2024 19:41:45 [INFO|DP=0|PP=11|TP=0|ip-26-0-173-202]: Local number of parameters: 41.9M (80.01MiB)
[default3]:07/02/2024 19:41:45 [INFO|DP=0|PP=11|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 81.02MiB. Peak allocated: 83.05MiB Peak reserved: 96.00MiB
[default3]:07/02/2024 19:41:45 [INFO|DP=0|PP=11|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default7]:07/02/2024 19:41:45 [INFO|DP=0|PP=15|TP=0|ip-26-0-173-202]: Local number of parameters: 0 (0.00MiB)
[default7]:07/02/2024 19:41:45 [INFO|DP=0|PP=15|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 0.01MiB. Peak allocated: 0.02MiB Peak reserved: 2.00MiB
[default7]:07/02/2024 19:41:45 [INFO|DP=0|PP=15|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default2]:07/02/2024 19:41:45 [INFO|DP=0|PP=10|TP=0|ip-26-0-173-202]: Local number of parameters: 83.9M (160.02MiB)
[default2]:07/02/2024 19:41:45 [INFO|DP=0|PP=10|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default2]:07/02/2024 19:41:45 [INFO|DP=0|PP=10|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=8|TP=0|ip-26-0-173-202]: Local number of parameters: 41.9M (80.01MiB)
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=8|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 81.02MiB. Peak allocated: 83.05MiB Peak reserved: 96.00MiB
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=8|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default6]:07/02/2024 19:41:45 [INFO|DP=0|PP=14|TP=0|ip-26-0-173-202]: Local number of parameters: 103M (196.32MiB)
[default6]:07/02/2024 19:41:45 [INFO|DP=0|PP=14|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 196.33MiB. Peak allocated: 196.34MiB Peak reserved: 200.00MiB
[default6]:07/02/2024 19:41:45 [INFO|DP=0|PP=14|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default4]:07/02/2024 19:41:45 [INFO|DP=0|PP=12|TP=0|ip-26-0-173-202]: Local number of parameters: 83.9M (160.02MiB)
[default4]:07/02/2024 19:41:45 [INFO|DP=0|PP=12|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default4]:07/02/2024 19:41:45 [INFO|DP=0|PP=12|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default5]:07/02/2024 19:41:45 [INFO|DP=0|PP=13|TP=0|ip-26-0-173-202]: Local number of parameters: 83.9M (160.02MiB)
[default5]:07/02/2024 19:41:45 [INFO|DP=0|PP=13|TP=0|ip-26-0-173-202]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default5]:07/02/2024 19:41:45 [INFO|DP=0|PP=13|TP=0|ip-26-0-173-202]: No checkpoint path provided.
[default4]:07/02/2024 19:41:45 [INFO|DP=0|PP=4|TP=0|ip-26-0-171-21]: Local number of parameters: 83.9M (160.02MiB)
[default4]:07/02/2024 19:41:45 [INFO|DP=0|PP=4|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default5]:07/02/2024 19:41:45 [INFO|DP=0|PP=5|TP=0|ip-26-0-171-21]: Local number of parameters: 41.9M (80.01MiB)
[default5]:07/02/2024 19:41:45 [INFO|DP=0|PP=5|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 81.02MiB. Peak allocated: 83.05MiB Peak reserved: 96.00MiB
[default5]:07/02/2024 19:41:45 [INFO|DP=0|PP=5|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default7]:07/02/2024 19:41:45 [INFO|DP=0|PP=7|TP=0|ip-26-0-171-21]: Local number of parameters: 83.9M (160.02MiB)
[default7]:07/02/2024 19:41:45 [INFO|DP=0|PP=7|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Total number of parameters: 1.21G (2312.82MiB)
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Local number of parameters: 187M (356.33MiB)
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 358.34MiB. Peak allocated: 360.37MiB Peak reserved: 368.00MiB
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default0]:07/02/2024 19:41:45 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Parametrizing model parameters using StandardParametrizator
[default6]:07/02/2024 19:41:45 [INFO|DP=0|PP=6|TP=0|ip-26-0-171-21]: Local number of parameters: 83.9M (160.02MiB)
[default6]:07/02/2024 19:41:45 [INFO|DP=0|PP=6|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default4]:07/02/2024 19:41:45 [INFO|DP=0|PP=4|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default2]:07/02/2024 19:41:45 [INFO|DP=0|PP=2|TP=0|ip-26-0-171-21]: Local number of parameters: 41.9M (80.01MiB)
[default1]:07/02/2024 19:41:45 [INFO|DP=0|PP=1|TP=0|ip-26-0-171-21]: Local number of parameters: 83.9M (160.02MiB)
[default1]:07/02/2024 19:41:45 [INFO|DP=0|PP=1|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default6]:07/02/2024 19:41:45 [INFO|DP=0|PP=6|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default3]:07/02/2024 19:41:45 [INFO|DP=0|PP=3|TP=0|ip-26-0-171-21]: Local number of parameters: 83.9M (160.02MiB)
[default1]:07/02/2024 19:41:45 [INFO|DP=0|PP=1|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default2]:07/02/2024 19:41:45 [INFO|DP=0|PP=2|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 81.02MiB. Peak allocated: 83.05MiB Peak reserved: 96.00MiB
[default2]:07/02/2024 19:41:45 [INFO|DP=0|PP=2|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default7]:07/02/2024 19:41:45 [INFO|DP=0|PP=7|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default3]:07/02/2024 19:41:45 [INFO|DP=0|PP=3|TP=0|ip-26-0-171-21]: [After model building] Memory usage: 162.03MiB. Peak allocated: 164.06MiB Peak reserved: 170.00MiB
[default3]:07/02/2024 19:41:45 [INFO|DP=0|PP=3|TP=0|ip-26-0-171-21]: No checkpoint path provided.
[default0]:07/02/2024 19:41:46 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [Optimizer Building] Using LearningRateForSP as learning rate
[default0]:07/02/2024 19:41:46 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [ZeRO sharding] Size of optimizer params per rank:
[default0]:07/02/2024 19:41:46 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [ZeRO sharding] DP Rank 0 has 187M out of 187M (100.00%) params' optimizer states
[default0]:07/02/2024 19:41:47 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [Training Plan] Stage Training Stage has 19 remaining training steps and has consumed 0 samples
[default0]:07/02/2024 19:41:47 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Using `datasets` library
[default0]:07/02/2024 19:41:47 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Loading tokenizer from openai-community/gpt2 and transformers/hf_hub versions ('4.41.2', '0.23.4')
[default0]:07/02/2024 19:41:47 [WARNING|DP=0|PP=0|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default0]:Repo card metadata block was not found. Setting CardData to empty.
[default0]:07/02/2024 19:41:48 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [Training Plan] There are 1 training stages
[default0]:07/02/2024 19:41:48 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [Stage Training Stage] start from step 1
[default0]:07/02/2024 19:41:48 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]:
[default0]:07/02/2024 19:41:48 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: [Start training] datetime: 2024-07-02 19:41:48.749555 | mbs: 32 | grad_accum: 32 | global_batch_size: 1024 | sequence_length: 4096 | train_steps: 20 | start_iteration_step: 0 | consumed_train_samples: 0
[default0]:07/02/2024 19:41:48 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Resuming training from stage Training Stage, it has trained for 0 samples and has 19 remaining train steps
[default0]:07/02/2024 19:41:48 [INFO|DP=0|PP=0|TP=0|ip-26-0-171-21]: Memory usage: 1783.67MiB. Peak allocated 1783.67MiB. Peak reserved: 1796.00MiB
[default5]:07/02/2024 19:41:48 [WARNING|DP=0|PP=5|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default5]:Repo card metadata block was not found. Setting CardData to empty.
[default3]:07/02/2024 19:41:48 [WARNING|DP=0|PP=11|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default7]:07/02/2024 19:41:48 [WARNING|DP=0|PP=15|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default4]:Repo card metadata block was not found. Setting CardData to empty.
[default3]:Repo card metadata block was not found. Setting CardData to empty.
[default5]:Repo card metadata block was not found. Setting CardData to empty.
[default1]:07/02/2024 19:41:48 [WARNING|DP=0|PP=9|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default0]:Repo card metadata block was not found. Setting CardData to empty.
[default2]:07/02/2024 19:41:48 [WARNING|DP=0|PP=10|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default0]:07/02/2024 19:41:48 [WARNING|DP=0|PP=8|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default4]:07/02/2024 19:41:48 [WARNING|DP=0|PP=12|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default6]:07/02/2024 19:41:48 [WARNING|DP=0|PP=14|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default5]:07/02/2024 19:41:48 [WARNING|DP=0|PP=13|TP=0|ip-26-0-173-202]: Repo card metadata block was not found. Setting CardData to empty.
[default1]:Repo card metadata block was not found. Setting CardData to empty.
[default2]:Repo card metadata block was not found. Setting CardData to empty.
[default7]:Repo card metadata block was not found. Setting CardData to empty.
[default6]:Repo card metadata block was not found. Setting CardData to empty.
[default4]:07/02/2024 19:41:48 [WARNING|DP=0|PP=4|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default3]:07/02/2024 19:41:48 [WARNING|DP=0|PP=3|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default1]:07/02/2024 19:41:48 [WARNING|DP=0|PP=1|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default6]:Repo card metadata block was not found. Setting CardData to empty.
[default7]:07/02/2024 19:41:48 [WARNING|DP=0|PP=7|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default6]:07/02/2024 19:41:48 [WARNING|DP=0|PP=6|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default2]:07/02/2024 19:41:48 [WARNING|DP=0|PP=2|TP=0|ip-26-0-171-21]: Repo card metadata block was not found. Setting CardData to empty.
[default4]:Repo card metadata block was not found. Setting CardData to empty.
[default7]:Repo card metadata block was not found. Setting CardData to empty.
[default1]:Repo card metadata block was not found. Setting CardData to empty.
[default2]:Repo card metadata block was not found. Setting CardData to empty.
[default3]:Repo card metadata block was not found. Setting CardData to empty.
[default0]:[rank0]: Traceback (most recent call last):
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default0]:[rank0]: trainer.train(dataloader)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default0]:[rank0]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default0]:[rank0]: outputs = self.pipeline_engine.train_batch_iter(
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default0]:[rank0]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default0]:[rank0]: output = model(**micro_batch)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank0]: return self._call_impl(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank0]: return forward_call(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default0]:[rank0]: sharded_logits = self.model(
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank0]: return self._call_impl(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank0]: return forward_call(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default0]:[rank0]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default0]:[rank0]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank0]: return self._call_impl(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank0]: return forward_call(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 151, in forward
[default0]:[rank0]: output = self.pp_block(**new_kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank0]: return self._call_impl(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank0]: return forward_call(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 631, in forward
[default0]:[rank0]: output = self.attn(hidden_states=hidden_states, sequence_mask=sequence_mask)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank0]: return self._call_impl(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank0]: return forward_call(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 360, in forward
[default0]:[rank0]: qkv_states = self.qkv_proj(
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank0]: return self._call_impl(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank0]: return forward_call(*args, **kwargs)
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/tensor_parallel/nn.py", line 87, in forward
[default0]:[rank0]: return column_linear(
[default0]:[rank0]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/tensor_parallel/functional.py", line 359, in column_linear
[default0]:[rank0]: return F.linear(input, weight, bias)
[default0]:[rank0]: torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1.50 GiB. GPU
[default6]:[rank6]: Traceback (most recent call last):
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default6]:[rank6]: trainer.train(dataloader)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default6]:[rank6]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default6]:[rank6]: outputs = self.pipeline_engine.train_batch_iter(
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default6]:[rank6]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default6]:[rank6]: output = model(**micro_batch)
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default6]:[rank6]: return self._call_impl(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default6]:[rank6]: return forward_call(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default6]:[rank6]: sharded_logits = self.model(
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default6]:[rank6]: return self._call_impl(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default6]:[rank6]: return forward_call(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default6]:[rank6]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default6]:[rank6]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default6]:[rank6]: return self._call_impl(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default6]:[rank6]: return forward_call(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default6]:[rank6]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default6]:[rank6]: pipeline_state.run_communication()
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default6]:[rank6]: recv_activation_tensor = recv_activation()
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default6]:[rank6]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default6]:[rank6]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default6]:[rank6]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default6]:[rank6]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default6]:[rank6]: dist.recv(
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default6]:[rank6]: return func(*args, **kwargs)
[default6]:[rank6]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default6]:[rank6]: pg.recv([tensor], group_src_rank, tag).wait()
[default6]:[rank6]: torch.distributed.DistBackendError: [6] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '5:6', but store->get('5:6') got error: Connection reset by peer
[default6]:[rank6]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default6]:[rank6]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7fe31612e897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default6]:[rank6]: frame #1: <unknown function> + 0x5b3a23e (0x7fe34fc4b23e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7fe34fc45c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7fe34fc45f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7fe34fc46fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fe34fbfb371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fe34fbfb371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fe34fbfb371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fe34fbfb371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7fe317408189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default6]:[rank6]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7fe31740f610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default6]:[rank6]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7fe31742e978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default6]:[rank6]: frame #12: <unknown function> + 0x5adc309 (0x7fe34fbed309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #13: <unknown function> + 0x5ae6f10 (0x7fe34fbf7f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #14: <unknown function> + 0x5ae6fa5 (0x7fe34fbf7fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #15: <unknown function> + 0x5124446 (0x7fe34f235446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #16: <unknown function> + 0x1acf4b8 (0x7fe34bbe04b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #17: <unknown function> + 0x5aee004 (0x7fe34fbff004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #18: <unknown function> + 0x5af36b5 (0x7fe34fc046b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank6]: frame #19: <unknown function> + 0xd2631e (0x7fe3627ee31e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default6]:[rank6]: frame #20: <unknown function> + 0x47def4 (0x7fe361f45ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default6]:[rank6]: frame #21: <unknown function> + 0x1445a6 (0x5583ec0f65a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #22: _PyObject_MakeTpCall + 0x26b (0x5583ec0efa6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #23: <unknown function> + 0x150866 (0x5583ec102866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x5583ec0eb142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #25: _PyFunction_Vectorcall + 0x6c (0x5583ec0f6a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #26: PyObject_Call + 0xbc (0x5583ec102f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x5583ec0e92b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #28: _PyFunction_Vectorcall + 0x6c (0x5583ec0f6a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x5583ec0e78fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #30: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x5583ec0e78fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #32: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x5583ec0e78fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #34: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x5583ec0e78fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x5583ec0eef50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #37: _PyObject_Call_Prepend + 0x69 (0x5583ec100c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #38: <unknown function> + 0x211239 (0x5583ec1c3239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #39: _PyObject_MakeTpCall + 0x26b (0x5583ec0efa6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x5583ec0eb3e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #41: _PyFunction_Vectorcall + 0x6c (0x5583ec0f6a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x5583ec0e6c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #43: _PyFunction_Vectorcall + 0x6c (0x5583ec0f6a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x5583ec0e78fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #45: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #46: PyObject_Call + 0xbc (0x5583ec102f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x5583ec0e92b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #48: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #49: PyObject_Call + 0xbc (0x5583ec102f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x5583ec0e92b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #51: _PyFunction_Vectorcall + 0x6c (0x5583ec0f6a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x5583ec0ef007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #53: _PyObject_Call_Prepend + 0x69 (0x5583ec100c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #54: <unknown function> + 0x211239 (0x5583ec1c3239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #55: PyObject_Call + 0x207 (0x5583ec103067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x5583ec0e92b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #57: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x5583ec0e78fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #59: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #60: PyObject_Call + 0xbc (0x5583ec102f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x5583ec0e92b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #62: <unknown function> + 0x150582 (0x5583ec102582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: frame #63: PyObject_Call + 0xbc (0x5583ec102f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank6]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default5]:[rank5]: Traceback (most recent call last):
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default5]:[rank5]: trainer.train(dataloader)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default5]:[rank5]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default5]:[rank5]: outputs = self.pipeline_engine.train_batch_iter(
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default5]:[rank5]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default5]:[rank5]: output = model(**micro_batch)
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank5]: return self._call_impl(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank5]: return forward_call(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default5]:[rank5]: sharded_logits = self.model(
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank5]: return self._call_impl(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank5]: return forward_call(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default5]:[rank5]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default5]:[rank5]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank5]: return self._call_impl(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank5]: return forward_call(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default5]:[rank5]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default5]:[rank5]: pipeline_state.run_communication()
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default5]:[rank5]: recv_activation_tensor = recv_activation()
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default5]:[rank5]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default5]:[rank5]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default5]:[rank5]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default5]:[rank5]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default5]:[rank5]: dist.recv(
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default5]:[rank5]: return func(*args, **kwargs)
[default5]:[rank5]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default5]:[rank5]: pg.recv([tensor], group_src_rank, tag).wait()
[default5]:[rank5]: torch.distributed.DistBackendError: [5] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '4:5', but store->get('4:5') got error: Connection reset by peer
[default5]:[rank5]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default5]:[rank5]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f06cf2cb897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default5]:[rank5]: frame #1: <unknown function> + 0x5b3a23e (0x7f0708de823e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f0708de2c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f0708de2f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f0708de3fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0708d98371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0708d98371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0708d98371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0708d98371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f06d05a5189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank5]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f06d05ac610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank5]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f06d05cb978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank5]: frame #12: <unknown function> + 0x5adc309 (0x7f0708d8a309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #13: <unknown function> + 0x5ae6f10 (0x7f0708d94f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f0708d94fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #15: <unknown function> + 0x5124446 (0x7f07083d2446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #16: <unknown function> + 0x1acf4b8 (0x7f0704d7d4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #17: <unknown function> + 0x5aee004 (0x7f0708d9c004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #18: <unknown function> + 0x5af36b5 (0x7f0708da16b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank5]: frame #19: <unknown function> + 0xd2631e (0x7f071b98b31e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default5]:[rank5]: frame #20: <unknown function> + 0x47def4 (0x7f071b0e2ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default5]:[rank5]: frame #21: <unknown function> + 0x1445a6 (0x56335dfbb5a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #22: _PyObject_MakeTpCall + 0x26b (0x56335dfb4a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #23: <unknown function> + 0x150866 (0x56335dfc7866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x56335dfb0142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #25: _PyFunction_Vectorcall + 0x6c (0x56335dfbba2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #26: PyObject_Call + 0xbc (0x56335dfc7f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x56335dfae2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #28: _PyFunction_Vectorcall + 0x6c (0x56335dfbba2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x56335dfac8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #30: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x56335dfac8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #32: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x56335dfac8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #34: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x56335dfac8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x56335dfb3f50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #37: _PyObject_Call_Prepend + 0x69 (0x56335dfc5c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #38: <unknown function> + 0x211239 (0x56335e088239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #39: _PyObject_MakeTpCall + 0x26b (0x56335dfb4a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x56335dfb03e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #41: _PyFunction_Vectorcall + 0x6c (0x56335dfbba2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x56335dfabc5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #43: _PyFunction_Vectorcall + 0x6c (0x56335dfbba2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x56335dfac8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #45: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #46: PyObject_Call + 0xbc (0x56335dfc7f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x56335dfae2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #48: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #49: PyObject_Call + 0xbc (0x56335dfc7f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x56335dfae2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #51: _PyFunction_Vectorcall + 0x6c (0x56335dfbba2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x56335dfb4007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #53: _PyObject_Call_Prepend + 0x69 (0x56335dfc5c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #54: <unknown function> + 0x211239 (0x56335e088239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #55: PyObject_Call + 0x207 (0x56335dfc8067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x56335dfae2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #57: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x56335dfac8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #59: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #60: PyObject_Call + 0xbc (0x56335dfc7f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x56335dfae2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #62: <unknown function> + 0x150582 (0x56335dfc7582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: frame #63: PyObject_Call + 0xbc (0x56335dfc7f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank5]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default7]:[rank7]: Traceback (most recent call last):
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default7]:[rank7]: trainer.train(dataloader)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default7]:[rank7]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default7]:[rank7]: outputs = self.pipeline_engine.train_batch_iter(
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default7]:[rank7]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default7]:[rank7]: output = model(**micro_batch)
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default7]:[rank7]: return self._call_impl(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default7]:[rank7]: return forward_call(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default7]:[rank7]: sharded_logits = self.model(
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default7]:[rank7]: return self._call_impl(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default7]:[rank7]: return forward_call(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default7]:[rank7]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default7]:[rank7]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default7]:[rank7]: return self._call_impl(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default7]:[rank7]: return forward_call(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default7]:[rank7]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default7]:[rank7]: pipeline_state.run_communication()
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default7]:[rank7]: recv_activation_tensor = recv_activation()
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default7]:[rank7]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default7]:[rank7]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default7]:[rank7]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default7]:[rank7]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default7]:[rank7]: dist.recv(
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default7]:[rank7]: return func(*args, **kwargs)
[default7]:[rank7]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default7]:[rank7]: pg.recv([tensor], group_src_rank, tag).wait()
[default7]:[rank7]: torch.distributed.DistBackendError: [7] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '6:7', but store->get('6:7') got error: Connection reset by peer
[default7]:[rank7]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default7]:[rank7]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f29fd4c6897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default7]:[rank7]: frame #1: <unknown function> + 0x5b3a23e (0x7f2a36fe323e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f2a36fddc87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f2a36fddf82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f2a36fdefd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2a36f93371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2a36f93371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2a36f93371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2a36f93371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f29fe7a0189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default7]:[rank7]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f29fe7a7610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default7]:[rank7]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f29fe7c6978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default7]:[rank7]: frame #12: <unknown function> + 0x5adc309 (0x7f2a36f85309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #13: <unknown function> + 0x5ae6f10 (0x7f2a36f8ff10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f2a36f8ffa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #15: <unknown function> + 0x5124446 (0x7f2a365cd446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #16: <unknown function> + 0x1acf4b8 (0x7f2a32f784b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #17: <unknown function> + 0x5aee004 (0x7f2a36f97004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #18: <unknown function> + 0x5af36b5 (0x7f2a36f9c6b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank7]: frame #19: <unknown function> + 0xd2631e (0x7f2a49b8631e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default7]:[rank7]: frame #20: <unknown function> + 0x47def4 (0x7f2a492ddef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default7]:[rank7]: frame #21: <unknown function> + 0x1445a6 (0x5636727f85a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #22: _PyObject_MakeTpCall + 0x26b (0x5636727f1a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #23: <unknown function> + 0x150866 (0x563672804866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x5636727ed142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #25: _PyFunction_Vectorcall + 0x6c (0x5636727f8a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #26: PyObject_Call + 0xbc (0x563672804f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x5636727eb2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #28: _PyFunction_Vectorcall + 0x6c (0x5636727f8a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x5636727e98fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #30: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x5636727e98fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #32: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x5636727e98fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #34: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x5636727e98fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x5636727f0f50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #37: _PyObject_Call_Prepend + 0x69 (0x563672802c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #38: <unknown function> + 0x211239 (0x5636728c5239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #39: _PyObject_MakeTpCall + 0x26b (0x5636727f1a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x5636727ed3e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #41: _PyFunction_Vectorcall + 0x6c (0x5636727f8a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x5636727e8c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #43: _PyFunction_Vectorcall + 0x6c (0x5636727f8a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x5636727e98fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #45: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #46: PyObject_Call + 0xbc (0x563672804f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x5636727eb2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #48: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #49: PyObject_Call + 0xbc (0x563672804f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x5636727eb2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #51: _PyFunction_Vectorcall + 0x6c (0x5636727f8a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x5636727f1007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #53: _PyObject_Call_Prepend + 0x69 (0x563672802c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #54: <unknown function> + 0x211239 (0x5636728c5239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #55: PyObject_Call + 0x207 (0x563672805067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x5636727eb2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #57: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x5636727e98fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #59: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #60: PyObject_Call + 0xbc (0x563672804f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x5636727eb2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #62: <unknown function> + 0x150582 (0x563672804582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: frame #63: PyObject_Call + 0xbc (0x563672804f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank7]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default5]:[rank13]: Traceback (most recent call last):
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default4]:[rank12]: Traceback (most recent call last):
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default4]:[rank12]: trainer.train(dataloader)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default4]:[rank12]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default3]:[rank11]: Traceback (most recent call last):
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default3]:[rank11]: trainer.train(dataloader)
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default3]:[rank11]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default4]:[rank12]: outputs = self.pipeline_engine.train_batch_iter(
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default5]:[rank13]: trainer.train(dataloader)
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default4]:[rank12]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default4]:[rank12]: output = model(**micro_batch)
[default3]:[rank11]: outputs = self.pipeline_engine.train_batch_iter(
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default3]:[rank11]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default4]:[rank12]: return self._call_impl(*args, **kwargs)
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default5]:[rank13]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default3]:[rank11]: output = model(**micro_batch)
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default3]:[rank11]: return self._call_impl(*args, **kwargs)
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default0]:[rank8]: Traceback (most recent call last):
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default4]:[rank12]: return forward_call(*args, **kwargs)
[default1]:[rank9]: Traceback (most recent call last):
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank13]: outputs = self.pipeline_engine.train_batch_iter(
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default1]:[rank9]: trainer.train(dataloader)
[default3]:[rank11]: return forward_call(*args, **kwargs)
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default5]:[rank13]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default0]:[rank8]: trainer.train(dataloader)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default4]:[rank12]: sharded_logits = self.model(
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default3]:[rank11]: sharded_logits = self.model(
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default1]:[rank9]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default5]:[rank13]: output = model(**micro_batch)
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default1]:[rank9]: outputs = self.pipeline_engine.train_batch_iter(
[default0]:[rank8]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default3]:[rank11]: return self._call_impl(*args, **kwargs)
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default4]:[rank12]: return self._call_impl(*args, **kwargs)
[default0]:[rank8]: outputs = self.pipeline_engine.train_batch_iter(
[default3]:[rank11]: return forward_call(*args, **kwargs)
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default0]:[rank8]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default1]:[rank9]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default4]:[rank12]: return forward_call(*args, **kwargs)
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default5]:[rank13]: return self._call_impl(*args, **kwargs)
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default3]:[rank11]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default3]:[rank11]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default4]:[rank12]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default3]:[rank11]: return self._call_impl(*args, **kwargs)
[default5]:[rank13]: return forward_call(*args, **kwargs)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default2]:[rank10]: Traceback (most recent call last):
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank8]: output = model(**micro_batch)
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default5]:[rank13]: sharded_logits = self.model(
[default0]:[rank8]: return self._call_impl(*args, **kwargs)
[default3]:[rank11]: return forward_call(*args, **kwargs)
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default4]:[rank12]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default2]:[rank10]: trainer.train(dataloader)
[default3]:[rank11]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default4]:[rank12]: return self._call_impl(*args, **kwargs)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default2]:[rank10]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank13]: return self._call_impl(*args, **kwargs)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default2]:[rank10]: outputs = self.pipeline_engine.train_batch_iter(
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default4]:[rank12]: return forward_call(*args, **kwargs)
[default3]:[rank11]: pipeline_state.run_communication()
[default0]:[rank8]: return forward_call(*args, **kwargs)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default3]:[rank11]: recv_activation_tensor = recv_activation()
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default5]:[rank13]: return forward_call(*args, **kwargs)
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default0]:[rank8]: sharded_logits = self.model(
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank8]: return self._call_impl(*args, **kwargs)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default2]:[rank10]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default3]:[rank11]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default1]:[rank9]: output = model(**micro_batch)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default1]:[rank9]: return self._call_impl(*args, **kwargs)
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default5]:[rank13]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default0]:[rank8]: return forward_call(*args, **kwargs)
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default2]:[rank10]: output = model(**micro_batch)
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default5]:[rank13]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default0]:[rank8]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default3]:[rank11]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default2]:[rank10]: return self._call_impl(*args, **kwargs)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default0]:[rank8]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default2]:[rank10]: return forward_call(*args, **kwargs)
[default3]:[rank11]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default3]:[rank11]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default1]:[rank9]: return forward_call(*args, **kwargs)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank13]: return self._call_impl(*args, **kwargs)
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank8]: return self._call_impl(*args, **kwargs)
[default4]:[rank12]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default1]:[rank9]: sharded_logits = self.model(
[default2]:[rank10]: sharded_logits = self.model(
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default3]:[rank11]: dist.recv(
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default3]:[rank11]: return func(*args, **kwargs)
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default2]:[rank10]: return self._call_impl(*args, **kwargs)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default0]:[rank8]: return forward_call(*args, **kwargs)
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default4]:[rank12]: pipeline_state.run_communication()
[default3]:[rank11]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default0]:[rank8]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default2]:[rank10]: return forward_call(*args, **kwargs)
[default3]:[rank11]: pg.recv([tensor], group_src_rank, tag).wait()
[default1]:[rank9]: return self._call_impl(*args, **kwargs)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank13]: return forward_call(*args, **kwargs)
[default3]:[rank11]: torch.distributed.DistBackendError: [11] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '10:11', but store->get('10:11') got error: Connection reset by peer
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default4]:[rank12]: recv_activation_tensor = recv_activation()
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default2]:[rank10]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default3]:[rank11]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default2]:[rank10]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default1]:[rank9]: return forward_call(*args, **kwargs)
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default1]:[rank9]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default5]:[rank13]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default2]:[rank10]: return self._call_impl(*args, **kwargs)
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default2]:[rank10]: return forward_call(*args, **kwargs)
[default3]:[rank11]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f542efc5897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default3]:[rank11]: frame #1: <unknown function> + 0x5b3a23e (0x7f5468ae223e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default4]:[rank12]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default7]:[rank15]: Traceback (most recent call last):
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 780, in forward_with_hidden_states
[default2]:[rank10]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default3]:[rank11]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f5468adcc87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default1]:[rank9]: hidden_encoder_states = encoder_block(**hidden_encoder_states)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default3]:[rank11]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f5468adcf82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f5468addfd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default2]:[rank10]: pipeline_state.run_communication()
[default6]:[rank14]: Traceback (most recent call last):
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py", line 237, in <module>
[default3]:[rank11]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f5468a92371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default5]:[rank13]: pipeline_state.run_communication()
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default7]:[rank15]: trainer.train(dataloader)
[default5]:[rank13]: recv_activation_tensor = recv_activation()
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default0]:[rank8]: pipeline_state.run_communication()
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default0]:[rank8]: recv_activation_tensor = recv_activation()
[default3]:[rank11]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f5468a92371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f5468a92371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f5468a92371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: trainer.train(dataloader)
[default1]:[rank9]: return self._call_impl(*args, **kwargs)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank13]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default1]:[rank9]: return forward_call(*args, **kwargs)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 429, in train
[default6]:[rank14]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default0]:[rank8]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default2]:[rank10]: recv_activation_tensor = recv_activation()
[default7]:[rank15]: outputs, loss_avg = self.training_step(dataloader=self.current_dataloader)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/trainer.py", line 462, in training_step
[default4]:[rank12]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default4]:[rank12]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default0]:[rank8]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default3]:[rank11]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f543029f189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default6]:[rank14]: outputs = self.pipeline_engine.train_batch_iter(
[default1]:[rank9]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default4]:[rank12]: dist.recv(
[default7]:[rank15]: outputs = self.pipeline_engine.train_batch_iter(
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default1]:[rank9]: pipeline_state.run_communication()
[default3]:[rank11]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f54302a6610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank13]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default5]:[rank13]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 252, in train_batch_iter
[default3]:[rank11]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f54302c5978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default2]:[rank10]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default5]:[rank13]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 278, in train_batch_iter
[default6]:[rank14]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default4]:[rank12]: return func(*args, **kwargs)
[default7]:[rank15]: output = self.forward(context=context, state=state, micro_batch=micro_batch, model=model)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default5]:[rank13]: dist.recv(
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default3]:[rank11]: frame #12: <unknown function> + 0x5adc309 (0x7f5468a84309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #13: <unknown function> + 0x5ae6f10 (0x7f5468a8ef10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default4]:[rank12]: pg.recv([tensor], group_src_rank, tag).wait()
[default0]:[rank8]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default2]:[rank10]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default5]:[rank13]: return func(*args, **kwargs)
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default0]:[rank8]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/engine.py", line 44, in forward
[default4]:[rank12]: torch.distributed.DistBackendError: [12] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '11:12', but store->get('11:12') got error: Connection reset by peer
[default4]:[rank12]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default1]:[rank9]: recv_activation_tensor = recv_activation()
[default3]:[rank11]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f5468a8efa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default4]:[rank12]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f942ad0c897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default4]:[rank12]: frame #1: <unknown function> + 0x5b3a23e (0x7f946482923e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default7]:[rank15]: output = model(**micro_batch)
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default7]:[rank15]: return self._call_impl(*args, **kwargs)
[default0]:[rank8]: dist.recv(
[default3]:[rank11]: frame #15: <unknown function> + 0x5124446 (0x7f54680cc446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default6]:[rank14]: output = model(**micro_batch)
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default4]:[rank12]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f9464823c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default6]:[rank14]: return self._call_impl(*args, **kwargs)
[default3]:[rank11]: frame #16: <unknown function> + 0x1acf4b8 (0x7f5464a774b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #17: <unknown function> + 0x5aee004 (0x7f5468a96004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f9464823f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: return func(*args, **kwargs)
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default3]:[rank11]: frame #18: <unknown function> + 0x5af36b5 (0x7f5468a9b6b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: pg.recv([tensor], group_src_rank, tag).wait()
[default5]:[rank13]: torch.distributed.DistBackendError: [13] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '12:13', but store->get('12:13') got error: Connection reset by peer
[default0]:[rank8]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default6]:[rank14]: return forward_call(*args, **kwargs)
[default7]:[rank15]: return forward_call(*args, **kwargs)
[default4]:[rank12]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f9464824fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default0]:[rank8]: pg.recv([tensor], group_src_rank, tag).wait()
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default7]:[rank15]: sharded_logits = self.model(
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default1]:[rank9]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default2]:[rank10]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default2]:[rank10]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default7]:[rank15]: return self._call_impl(*args, **kwargs)
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default3]:[rank11]: frame #19: <unknown function> + 0xd2631e (0x7f547b68531e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default3]:[rank11]: frame #20: <unknown function> + 0x47def4 (0x7f547addcef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default5]:[rank13]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default1]:[rank9]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default2]:[rank10]: dist.recv(
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 891, in forward
[default5]:[rank13]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7fdb3adad897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default4]:[rank12]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f94647d9371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default3]:[rank11]: frame #21: <unknown function> + 0x1445a6 (0x563f5ee605a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #22: _PyObject_MakeTpCall + 0x26b (0x563f5ee59a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #1: <unknown function> + 0x5b3a23e (0x7fdb748ca23e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default2]:[rank10]: return func(*args, **kwargs)
[default7]:[rank15]: return forward_call(*args, **kwargs)
[default0]:[rank8]: torch.distributed.DistBackendError: [8] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '7:8', but store->get('7:8') got error: Connection reset by peer
[default1]:[rank9]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default5]:[rank13]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7fdb748c4c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default4]:[rank12]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f94647d9371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default1]:[rank9]: dist.recv(
[default5]:[rank13]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7fdb748c4f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7fdb748c5fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default3]:[rank11]: frame #23: <unknown function> + 0x150866 (0x563f5ee6c866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: pg.recv([tensor], group_src_rank, tag).wait()
[default2]:[rank10]: torch.distributed.DistBackendError: [10] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '9:10', but store->get('9:10') got error: Connection reset by peer
[default0]:[rank8]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f42d5e99897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default7]:[rank15]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default5]:[rank13]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fdb7487a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default0]:[rank8]: frame #1: <unknown function> + 0x5b3a23e (0x7f430f9b623e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x563f5ee55142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f94647d9371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f94647d9371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: return func(*args, **kwargs)
[default6]:[rank14]: sharded_logits = self.model(
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default3]:[rank11]: frame #25: _PyFunction_Vectorcall + 0x6c (0x563f5ee60a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #26: PyObject_Call + 0xbc (0x563f5ee6cf1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 786, in forward_with_hidden_states
[default7]:[rank15]: fp32_sharded_logits = self.cast_to_fp32(x=sharded_logits)["output"]
[default6]:[rank14]: return self._call_impl(*args, **kwargs)
[default4]:[rank12]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f942bfe6189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default0]:[rank8]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f430f9b0c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x563f5ee532b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #28: _PyFunction_Vectorcall + 0x6c (0x563f5ee60a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default7]:[rank15]: return self._call_impl(*args, **kwargs)
[default5]:[rank13]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fdb7487a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default4]:[rank12]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f942bfed610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default5]:[rank13]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fdb7487a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7fdb7487a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f430f9b0f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x563f5ee518fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default4]:[rank12]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f942c00c978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default2]:[rank10]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default3]:[rank11]: frame #30: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x563f5ee518fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: return forward_call(*args, **kwargs)
[default0]:[rank8]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f430f9b1fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f430f966371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: pg.recv([tensor], group_src_rank, tag).wait()
[default4]:[rank12]: frame #12: <unknown function> + 0x5adc309 (0x7f94647cb309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #13: <unknown function> + 0x5ae6f10 (0x7f94647d5f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: return forward_call(*args, **kwargs)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default7]:[rank15]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default3]:[rank11]: frame #32: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x563f5ee518fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 764, in forward
[default6]:[rank14]: return self.forward_with_hidden_states(input_ids=input_ids, input_mask=input_mask)[0]
[default0]:[rank8]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f430f966371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f430f966371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7fdb3c087189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank13]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7fdb3c08e610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default0]:[rank8]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f430f966371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f94647d5fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #15: <unknown function> + 0x5124446 (0x7f9463e13446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7fdb3c0ad978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default7]:[rank15]: pipeline_state.run_communication()
[default2]:[rank10]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f2bf4a86897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default2]:[rank10]: frame #1: <unknown function> + 0x5b3a23e (0x7f2c2e5a323e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: torch.distributed.DistBackendError: [9] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '8:9', but store->get('8:9') got error: Connection reset by peer
[default1]:[rank9]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/models/llama.py", line 782, in forward_with_hidden_states
[default4]:[rank12]: frame #16: <unknown function> + 0x1acf4b8 (0x7f94607be4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #17: <unknown function> + 0x5aee004 (0x7f94647dd004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #12: <unknown function> + 0x5adc309 (0x7fdb7486c309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f6d90129897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default0]:[rank8]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f42d7173189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default3]:[rank11]: frame #34: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default7]:[rank15]: recv_activation_tensor = recv_activation()
[default1]:[rank9]: frame #1: <unknown function> + 0x5b3a23e (0x7f6dc9c4623e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f42d717a610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default3]:[rank11]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x563f5ee518fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f2c2e59dc87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f6dc9c40c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #13: <unknown function> + 0x5ae6f10 (0x7fdb74876f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: hidden_states = self.final_layer_norm(input=hidden_encoder_states["hidden_states"])["hidden_states"]
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[default6]:[rank14]: return self._call_impl(*args, **kwargs)
[default2]:[rank10]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f2c2e59df82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f42d7199978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default3]:[rank11]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x563f5ee58f50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #18: <unknown function> + 0x5af36b5 (0x7f94647e26b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default4]:[rank12]: frame #19: <unknown function> + 0xd2631e (0x7f94773cc31e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default4]:[rank12]: frame #20: <unknown function> + 0x47def4 (0x7f9476b23ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default6]:[rank14]: return forward_call(*args, **kwargs)
[default5]:[rank13]: frame #14: <unknown function> + 0x5ae6fa5 (0x7fdb74876fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default2]:[rank10]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f2c2e59efd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #21: <unknown function> + 0x1445a6 (0x55ecebf1d5a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #37: _PyObject_Call_Prepend + 0x69 (0x563f5ee6ac39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #15: <unknown function> + 0x5124446 (0x7fdb73eb4446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/block.py", line 126, in forward
[default1]:[rank9]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f6dc9c40f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #12: <unknown function> + 0x5adc309 (0x7f430f958309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f6dc9c41fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default3]:[rank11]: frame #38: <unknown function> + 0x211239 (0x563f5ef2d239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2c2e553371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f6dc9bf6371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #16: <unknown function> + 0x1acf4b8 (0x7fdb7085f4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #22: _PyObject_MakeTpCall + 0x26b (0x55ecebf16a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: new_kwargs[name] = recv_from_pipeline_state_buffer(
[default1]:[rank9]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f6dc9bf6371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f6dc9bf6371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #13: <unknown function> + 0x5ae6f10 (0x7f430f962f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2c2e553371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/functional.py", line 117, in recv_from_pipeline_state_buffer
[default6]:[rank14]: pipeline_state.run_communication()
[default1]:[rank9]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f6dc9bf6371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f430f962fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #15: <unknown function> + 0x5124446 (0x7f430efa0446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #16: <unknown function> + 0x1acf4b8 (0x7f430b94b4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 150, in run_communication
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default5]:[rank13]: frame #17: <unknown function> + 0x5aee004 (0x7fdb7487e004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #18: <unknown function> + 0x5af36b5 (0x7fdb748836b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2c2e553371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f6d91403189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default4]:[rank12]: frame #23: <unknown function> + 0x150866 (0x55ecebf29866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #17: <unknown function> + 0x5aee004 (0x7f430f96a004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: recv_activation_tensor = recv_activation()
[default1]:[rank9]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f6d9140a610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default1]:[rank9]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f6d91429978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank13]: frame #19: <unknown function> + 0xd2631e (0x7fdb8746d31e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default5]:[rank13]: frame #20: <unknown function> + 0x47def4 (0x7fdb86bc4ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default7]:[rank15]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default7]:[rank15]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default0]:[rank8]: frame #18: <unknown function> + 0x5af36b5 (0x7f430f96f6b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f2c2e553371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/state.py", line 31, in __call__
[default5]:[rank13]: frame #21: <unknown function> + 0x1445a6 (0x55f54d67c5a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #12: <unknown function> + 0x5adc309 (0x7f6dc9be8309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f2bf5d60189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default6]:[rank14]: return self.p2p.recv_tensors(num_tensors=1, from_rank=self.from_rank)[0]
[default0]:[rank8]: frame #19: <unknown function> + 0xd2631e (0x7f432255931e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default1]:[rank9]: frame #13: <unknown function> + 0x5ae6f10 (0x7f6dc9bf2f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x55ecebf12142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #25: _PyFunction_Vectorcall + 0x6c (0x55ecebf1da2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #22: _PyObject_MakeTpCall + 0x26b (0x55f54d675a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #39: _PyObject_MakeTpCall + 0x26b (0x563f5ee59a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f6dc9bf2fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: dist.recv(
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default5]:[rank13]: frame #23: <unknown function> + 0x150866 (0x55f54d688866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x55f54d671142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 353, in recv_tensors
[default4]:[rank12]: frame #26: PyObject_Call + 0xbc (0x55ecebf29f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: return func(*args, **kwargs)
[default3]:[rank11]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x563f5ee553e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #15: <unknown function> + 0x5124446 (0x7f6dc9230446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: buffers, futures = self.irecv_tensors(num_tensors=num_tensors, from_rank=from_rank, tag=tag)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 326, in irecv_tensors
[default7]:[rank15]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default7]:[rank15]: pg.recv([tensor], group_src_rank, tag).wait()
[default2]:[rank10]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f2bf5d67610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default6]:[rank14]: meta = self._recv_meta(from_rank=from_rank, tag=tag)
[default0]:[rank8]: frame #20: <unknown function> + 0x47def4 (0x7f4321cb0ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default3]:[rank11]: frame #41: _PyFunction_Vectorcall + 0x6c (0x563f5ee60a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #25: _PyFunction_Vectorcall + 0x6c (0x55f54d67ca2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #16: <unknown function> + 0x1acf4b8 (0x7f6dc5bdb4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: File "/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/src/nanotron/parallel/pipeline_parallel/p2p.py", line 246, in _recv_meta
[default0]:[rank8]: frame #21: <unknown function> + 0x1445a6 (0x564eb4d235a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f2bf5d86978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default2]:[rank10]: frame #12: <unknown function> + 0x5adc309 (0x7f2c2e545309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: torch.distributed.DistBackendError: [15] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '14:15', but store->get('14:15') got error: Connection reset by peer
[default3]:[rank11]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x563f5ee50c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #22: _PyObject_MakeTpCall + 0x26b (0x564eb4d1ca6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #23: <unknown function> + 0x150866 (0x564eb4d2f866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #13: <unknown function> + 0x5ae6f10 (0x7f2c2e54ff10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #43: _PyFunction_Vectorcall + 0x6c (0x563f5ee60a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x55ecebf102b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x564eb4d18142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #26: PyObject_Call + 0xbc (0x55f54d688f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f2c2e54ffa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #15: <unknown function> + 0x5124446 (0x7f2c2db8d446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default7]:[rank15]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f3f2bacc897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default3]:[rank11]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x563f5ee518fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #28: _PyFunction_Vectorcall + 0x6c (0x55ecebf1da2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x55ecebf0e8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #25: _PyFunction_Vectorcall + 0x6c (0x564eb4d23a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #17: <unknown function> + 0x5aee004 (0x7f6dc9bfa004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #18: <unknown function> + 0x5af36b5 (0x7f6dc9bff6b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #19: <unknown function> + 0xd2631e (0x7f6ddc7e931e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default5]:[rank13]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x55f54d66f2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #45: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #46: PyObject_Call + 0xbc (0x563f5ee6cf1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x563f5ee532b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #1: <unknown function> + 0x5b3a23e (0x7f3f655e923e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #26: PyObject_Call + 0xbc (0x564eb4d2ff1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x564eb4d162b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #28: _PyFunction_Vectorcall + 0x6c (0x564eb4d23a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #30: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: dist.recv(
[default2]:[rank10]: frame #16: <unknown function> + 0x1acf4b8 (0x7f2c2a5384b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #17: <unknown function> + 0x5aee004 (0x7f2c2e557004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #48: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x55ecebf0e8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #32: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x55ecebf0e8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #20: <unknown function> + 0x47def4 (0x7f6ddbf40ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default1]:[rank9]: frame #21: <unknown function> + 0x1445a6 (0x5594a0c445a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #49: PyObject_Call + 0xbc (0x563f5ee6cf1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x563f5ee532b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #34: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f3f655e3c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x564eb4d148fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #51: _PyFunction_Vectorcall + 0x6c (0x563f5ee60a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x55ecebf0e8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #30: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/c10d_logger.py", line 75, in wrapper
[default4]:[rank12]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x55ecebf15f50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #18: <unknown function> + 0x5af36b5 (0x7f2c2e55c6b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x564eb4d148fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x563f5ee59007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #22: _PyObject_MakeTpCall + 0x26b (0x5594a0c3da6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #37: _PyObject_Call_Prepend + 0x69 (0x55ecebf27c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #28: _PyFunction_Vectorcall + 0x6c (0x55f54d67ca2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #32: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f3f655e3f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f3f655e4fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #19: <unknown function> + 0xd2631e (0x7f2c4114631e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default6]:[rank14]: return func(*args, **kwargs)
[default6]:[rank14]: File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/distributed_c10d.py", line 1932, in recv
[default0]:[rank8]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x564eb4d148fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #38: <unknown function> + 0x211239 (0x55ecebfea239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x55f54d66d8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #23: <unknown function> + 0x150866 (0x5594a0c50866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #34: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #39: _PyObject_MakeTpCall + 0x26b (0x55ecebf16a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #30: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x5594a0c39142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #53: _PyObject_Call_Prepend + 0x69 (0x563f5ee6ac39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x564eb4d148fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: pg.recv([tensor], group_src_rank, tag).wait()
[default2]:[rank10]: frame #20: <unknown function> + 0x47def4 (0x7f2c4089def4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default3]:[rank11]: frame #54: <unknown function> + 0x211239 (0x563f5ef2d239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x564eb4d1bf50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f3f65599371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f3f65599371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #21: <unknown function> + 0x1445a6 (0x5569379845a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #55: PyObject_Call + 0x207 (0x563f5ee6d067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #25: _PyFunction_Vectorcall + 0x6c (0x5594a0c44a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x55ecebf123e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x55f54d66d8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #37: _PyObject_Call_Prepend + 0x69 (0x564eb4d2dc39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x563f5ee532b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #26: PyObject_Call + 0xbc (0x5594a0c50f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f3f65599371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f3f65599371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: torch.distributed.DistBackendError: [14] is setting up NCCL communicator and retrieving ncclUniqueId from [0] via c10d key-value store by key '13:14', but store->get('13:14') got error: Connection reset by peer
[default3]:[rank11]: frame #57: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x5594a0c372b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #41: _PyFunction_Vectorcall + 0x6c (0x55ecebf1da2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: Exception raised from recvBytes at ../torch/csrc/distributed/c10d/Utils.hpp:672 (most recent call first):
[default6]:[rank14]: frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x57 (0x7f09da73d897 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libc10.so)
[default6]:[rank14]: frame #1: <unknown function> + 0x5b3a23e (0x7f0a1425a23e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #28: _PyFunction_Vectorcall + 0x6c (0x5594a0c44a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x563f5ee518fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #22: _PyObject_MakeTpCall + 0x26b (0x55693797da6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f3f2cda6189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank13]: frame #32: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #2: c10d::TCPStore::doWait(c10::ArrayRef<std::string>, std::chrono::duration<long, std::ratio<1l, 1000l> >) + 0x2c7 (0x7f0a14254c87 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default4]:[rank12]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x55ecebf0dc5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #59: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #23: <unknown function> + 0x150866 (0x556937990866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x5594a0c358fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #43: _PyFunction_Vectorcall + 0x6c (0x55ecebf1da2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x55ecebf0e8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x55f54d66d8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #38: <unknown function> + 0x211239 (0x564eb4df0239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f3f2cdad610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default2]:[rank10]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x556937979142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #45: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #34: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x55f54d66d8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #3: c10d::TCPStore::doGet(std::string const&) + 0x32 (0x7f0a14254f82 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: frame #4: c10d::TCPStore::get(std::string const&) + 0xa1 (0x7f0a14255fd1 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default3]:[rank11]: frame #60: PyObject_Call + 0xbc (0x563f5ee6cf1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x563f5ee532b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #30: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #39: _PyObject_MakeTpCall + 0x26b (0x564eb4d1ca6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f3f2cdcc978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default2]:[rank10]: frame #25: _PyFunction_Vectorcall + 0x6c (0x556937984a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #46: PyObject_Call + 0xbc (0x55ecebf29f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #5: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0a1420a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x55f54d674f50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #37: _PyObject_Call_Prepend + 0x69 (0x55f54d686c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x5594a0c358fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x564eb4d183e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #12: <unknown function> + 0x5adc309 (0x7f3f6558b309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default2]:[rank10]: frame #26: PyObject_Call + 0xbc (0x556937990f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #38: <unknown function> + 0x211239 (0x55f54d749239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #41: _PyFunction_Vectorcall + 0x6c (0x564eb4d23a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #13: <unknown function> + 0x5ae6f10 (0x7f3f65595f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: frame #6: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0a1420a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: frame #7: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0a1420a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x564eb4d13c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #32: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #62: <unknown function> + 0x150582 (0x563f5ee6c582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x55ecebf102b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #48: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #43: _PyFunction_Vectorcall + 0x6c (0x564eb4d23a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #39: _PyObject_MakeTpCall + 0x26b (0x55f54d675a6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x55f54d6713e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f3f65595fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: frame #15: <unknown function> + 0x5124446 (0x7f3f64bd3446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x564eb4d148fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x5569379772b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x5594a0c358fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: frame #63: PyObject_Call + 0xbc (0x563f5ee6cf1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #16: <unknown function> + 0x1acf4b8 (0x7f3f6157e4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #41: _PyFunction_Vectorcall + 0x6c (0x55f54d67ca2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #28: _PyFunction_Vectorcall + 0x6c (0x556937984a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #8: c10d::PrefixStore::get(std::string const&) + 0x31 (0x7f0a1420a371 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: frame #9: c10d::ProcessGroupNCCL::broadcastUniqueNCCLID(ncclUniqueId*, bool, std::string const&, int) + 0xa9 (0x7f09dba17189 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default4]:[rank12]: frame #49: PyObject_Call + 0xbc (0x55ecebf29f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x55f54d66cc5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #43: _PyFunction_Vectorcall + 0x6c (0x55f54d67ca2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x5569379758fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #30: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x55ecebf102b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x5569379758fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #32: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #45: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x55f54d66d8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default3]:[rank11]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default2]:[rank10]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x5569379758fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #10: c10d::ProcessGroupNCCL::getNCCLComm(std::string const&, c10::Device&, c10d::OpType, int, bool) + 0xc50 (0x7f09dba1e610 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank13]: frame #45: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #46: PyObject_Call + 0xbc (0x564eb4d2ff1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #34: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #51: _PyFunction_Vectorcall + 0x6c (0x55ecebf1da2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x55ecebf16007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #46: PyObject_Call + 0xbc (0x55f54d688f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x55f54d66f2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #17: <unknown function> + 0x5aee004 (0x7f3f6559d004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x564eb4d162b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #11: c10d::ProcessGroupNCCL::recv(std::vector<at::Tensor, std::allocator<at::Tensor> >&, int, int) + 0x5f8 (0x7f09dba3d978 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so)
[default5]:[rank13]: frame #48: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #18: <unknown function> + 0x5af36b5 (0x7f3f655a26b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: frame #19: <unknown function> + 0xd2631e (0x7f3f7818c31e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default7]:[rank15]: frame #20: <unknown function> + 0x47def4 (0x7f3f778e3ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default6]:[rank14]: frame #12: <unknown function> + 0x5adc309 (0x7f0a141fc309 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x5594a0c358fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #48: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #21: <unknown function> + 0x1445a6 (0x565006e825a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #34: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x5594a0c3cf50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #49: PyObject_Call + 0xbc (0x564eb4d2ff1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #49: PyObject_Call + 0xbc (0x55f54d688f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #22: _PyObject_MakeTpCall + 0x26b (0x565006e7ba6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #37: _PyObject_Call_Prepend + 0x69 (0x5594a0c4ec39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #53: _PyObject_Call_Prepend + 0x69 (0x55ecebf27c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x5569379758fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #13: <unknown function> + 0x5ae6f10 (0x7f0a14206f10 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default5]:[rank13]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x55f54d66f2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #51: _PyFunction_Vectorcall + 0x6c (0x55f54d67ca2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x564eb4d162b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #51: _PyFunction_Vectorcall + 0x6c (0x564eb4d23a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x55693797cf50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #38: <unknown function> + 0x211239 (0x5594a0d11239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #23: <unknown function> + 0x150866 (0x565006e8e866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x564eb4d1c007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #54: <unknown function> + 0x211239 (0x55ecebfea239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x55f54d675007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #14: <unknown function> + 0x5ae6fa5 (0x7f0a14206fa5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default6]:[rank14]: frame #15: <unknown function> + 0x5124446 (0x7f0a13844446 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #39: _PyObject_MakeTpCall + 0x26b (0x5594a0c3da6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #37: _PyObject_Call_Prepend + 0x69 (0x55693798ec39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #53: _PyObject_Call_Prepend + 0x69 (0x564eb4d2dc39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #16: <unknown function> + 0x1acf4b8 (0x7f0a101ef4b8 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default1]:[rank9]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x5594a0c393e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #55: PyObject_Call + 0x207 (0x55ecebf2a067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #54: <unknown function> + 0x211239 (0x564eb4df0239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #53: _PyObject_Call_Prepend + 0x69 (0x55f54d686c39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #41: _PyFunction_Vectorcall + 0x6c (0x5594a0c44a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x565006e77142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #55: PyObject_Call + 0x207 (0x564eb4d30067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #54: <unknown function> + 0x211239 (0x55f54d749239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x5594a0c34c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #25: _PyFunction_Vectorcall + 0x6c (0x565006e82a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x564eb4d162b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #55: PyObject_Call + 0x207 (0x55f54d689067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x55ecebf102b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #17: <unknown function> + 0x5aee004 (0x7f0a1420e004 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default7]:[rank15]: frame #26: PyObject_Call + 0xbc (0x565006e8ef1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x55f54d66f2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #57: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #57: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #43: _PyFunction_Vectorcall + 0x6c (0x5594a0c44a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #57: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x565006e752b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x564eb4d148fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x5594a0c358fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x55f54d66d8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #38: <unknown function> + 0x211239 (0x556937a51239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #59: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #45: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #46: PyObject_Call + 0xbc (0x5594a0c50f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x5594a0c372b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #18: <unknown function> + 0x5af36b5 (0x7f0a142136b5 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so)
[default0]:[rank8]: frame #60: PyObject_Call + 0xbc (0x564eb4d2ff1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x564eb4d162b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #39: _PyObject_MakeTpCall + 0x26b (0x55693797da6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #48: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #59: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #28: _PyFunction_Vectorcall + 0x6c (0x565006e82a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #19: <unknown function> + 0xd2631e (0x7f0a26dfd31e in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default4]:[rank12]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x55ecebf0e8fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #60: PyObject_Call + 0xbc (0x55f54d688f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x565006e738fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #20: <unknown function> + 0x47def4 (0x7f0a26554ef4 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/lib/libtorch_python.so)
[default2]:[rank10]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x5569379793e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #30: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #62: <unknown function> + 0x150582 (0x564eb4d2f582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #21: <unknown function> + 0x1445a6 (0x557efaf055a6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #49: PyObject_Call + 0xbc (0x5594a0c50f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x5594a0c372b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #51: _PyFunction_Vectorcall + 0x6c (0x5594a0c44a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: frame #63: PyObject_Call + 0xbc (0x564eb4d2ff1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default0]:[rank8]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default4]:[rank12]: frame #59: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #22: _PyObject_MakeTpCall + 0x26b (0x557efaefea6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #41: _PyFunction_Vectorcall + 0x6c (0x556937984a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x55f54d66f2b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #23: <unknown function> + 0x150866 (0x557efaf11866 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #60: PyObject_Call + 0xbc (0x55ecebf29f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x556937974c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x5594a0c3d007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #24: _PyEval_EvalFrameDefault + 0x4c12 (0x557efaefa142 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #25: _PyFunction_Vectorcall + 0x6c (0x557efaf05a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x565006e738fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x55ecebf102b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #43: _PyFunction_Vectorcall + 0x6c (0x556937984a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #53: _PyObject_Call_Prepend + 0x69 (0x5594a0c4ec39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #62: <unknown function> + 0x150582 (0x55f54d688582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #26: PyObject_Call + 0xbc (0x557efaf11f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #32: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: frame #63: PyObject_Call + 0xbc (0x55f54d688f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default5]:[rank13]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default1]:[rank9]: frame #54: <unknown function> + 0x211239 (0x5594a0d11239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x565006e738fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #27: _PyEval_EvalFrameDefault + 0x2d83 (0x557efaef82b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #55: PyObject_Call + 0x207 (0x5594a0c51067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x5569379758fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #28: _PyFunction_Vectorcall + 0x6c (0x557efaf05a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #29: _PyEval_EvalFrameDefault + 0x13ca (0x557efaef68fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #34: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x5594a0c372b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #45: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #46: PyObject_Call + 0xbc (0x556937990f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x565006e738fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #30: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x5569379772b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #48: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #31: _PyEval_EvalFrameDefault + 0x13ca (0x557efaef68fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #62: <unknown function> + 0x150582 (0x55ecebf29582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x565006e7af50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #49: PyObject_Call + 0xbc (0x556937990f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #32: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #37: _PyObject_Call_Prepend + 0x69 (0x565006e8cc39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x5569379772b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #33: _PyEval_EvalFrameDefault + 0x13ca (0x557efaef68fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: frame #63: PyObject_Call + 0xbc (0x55ecebf29f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #51: _PyFunction_Vectorcall + 0x6c (0x556937984a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #34: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #38: <unknown function> + 0x211239 (0x565006f4f239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #35: _PyEval_EvalFrameDefault + 0x13ca (0x557efaef68fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default4]:[rank12]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default1]:[rank9]: frame #57: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x5594a0c358fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #39: _PyObject_MakeTpCall + 0x26b (0x565006e7ba6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x565006e773e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #59: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #41: _PyFunction_Vectorcall + 0x6c (0x565006e82a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x565006e72c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #60: PyObject_Call + 0xbc (0x5594a0c50f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #43: _PyFunction_Vectorcall + 0x6c (0x565006e82a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #36: _PyObject_FastCallDictTstate + 0xd0 (0x557efaefdf50 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x55693797d007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x5594a0c372b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x565006e738fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #37: _PyObject_Call_Prepend + 0x69 (0x557efaf0fc39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #53: _PyObject_Call_Prepend + 0x69 (0x55693798ec39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #62: <unknown function> + 0x150582 (0x5594a0c50582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #45: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #38: <unknown function> + 0x211239 (0x557efafd2239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #54: <unknown function> + 0x211239 (0x556937a51239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: frame #63: PyObject_Call + 0xbc (0x5594a0c50f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #46: PyObject_Call + 0xbc (0x565006e8ef1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #39: _PyObject_MakeTpCall + 0x26b (0x557efaefea6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default1]:[rank9]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default7]:[rank15]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x565006e752b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #48: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #40: _PyEval_EvalFrameDefault + 0x4eb6 (0x557efaefa3e6 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #41: _PyFunction_Vectorcall + 0x6c (0x557efaf05a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #49: PyObject_Call + 0xbc (0x565006e8ef1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x565006e752b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #42: _PyEval_EvalFrameDefault + 0x72c (0x557efaef5c5c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #55: PyObject_Call + 0x207 (0x556937991067 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #51: _PyFunction_Vectorcall + 0x6c (0x565006e82a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x565006e7b007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #56: _PyEval_EvalFrameDefault + 0x2d83 (0x5569379772b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #43: _PyFunction_Vectorcall + 0x6c (0x557efaf05a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #57: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #44: _PyEval_EvalFrameDefault + 0x13ca (0x557efaef68fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #53: _PyObject_Call_Prepend + 0x69 (0x565006e8cc39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #54: <unknown function> + 0x211239 (0x565006f4f239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #45: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x5569379758fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #59: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #46: PyObject_Call + 0xbc (0x557efaf11f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #47: _PyEval_EvalFrameDefault + 0x2d83 (0x557efaef82b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #55: _PyObject_MakeTpCall + 0x26b (0x565006e7ba6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #60: PyObject_Call + 0xbc (0x556937990f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #48: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x5569379772b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #62: <unknown function> + 0x150582 (0x556937990582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #56: _PyEval_EvalFrameDefault + 0x5723 (0x565006e77c53 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: frame #63: PyObject_Call + 0xbc (0x556937990f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #49: PyObject_Call + 0xbc (0x557efaf11f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default2]:[rank10]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default6]:[rank14]: frame #50: _PyEval_EvalFrameDefault + 0x2d83 (0x557efaef82b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #51: _PyFunction_Vectorcall + 0x6c (0x557efaf05a2c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #52: _PyObject_FastCallDictTstate + 0x187 (0x557efaefe007 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #53: _PyObject_Call_Prepend + 0x69 (0x557efaf0fc39 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #54: <unknown function> + 0x211239 (0x557efafd2239 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #55: _PyObject_MakeTpCall + 0x26b (0x557efaefea6b in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #56: _PyEval_EvalFrameDefault + 0x5723 (0x557efaefac53 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #57: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x557efaef68fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #59: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #60: PyObject_Call + 0xbc (0x557efaf11f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x557efaef82b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #62: <unknown function> + 0x150582 (0x557efaf11582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: frame #63: PyObject_Call + 0xbc (0x557efaf11f1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #57: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #58: _PyEval_EvalFrameDefault + 0x13ca (0x565006e738fa in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #59: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #60: PyObject_Call + 0xbc (0x565006e8ef1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default6]:[rank14]: . This may indicate a possible application crash on rank 0 or a network set up issue.
[default7]:[rank15]: frame #61: _PyEval_EvalFrameDefault + 0x2d83 (0x565006e752b3 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #62: <unknown function> + 0x150582 (0x565006e8e582 in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: frame #63: PyObject_Call + 0xbc (0x565006e8ef1c in /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10)
[default7]:[rank15]: . This may indicate a possible application crash on rank 0 or a network set up issue.
W0702 19:42:02.175000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579549 closing signal SIGTERM
W0702 19:42:02.175000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579550 closing signal SIGTERM
W0702 19:42:02.176000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579551 closing signal SIGTERM
W0702 19:42:02.176000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579552 closing signal SIGTERM
W0702 19:42:02.177000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579553 closing signal SIGTERM
W0702 19:42:02.177000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579554 closing signal SIGTERM
W0702 19:42:02.178000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:851] Sending process 2579555 closing signal SIGTERM
E0702 19:42:04.401000 140644320167744 torch/distributed/elastic/multiprocessing/api.py:826] failed (exitcode: 1) local_rank: 0 (pid: 2579548) of binary: /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10
Traceback (most recent call last):
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/torchrun", line 8, in <module>
sys.exit(main())
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/elastic/multiprocessing/errors/__init__.py", line 347, in wrapper
return f(*args, **kwargs)
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/run.py", line 879, in main
run(args)
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/run.py", line 870, in run
elastic_launch(
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 132, in __call__
return launch_agent(self._config, self._entrypoint, list(args))
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 263, in launch_agent
raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError:
============================================================
/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py FAILED
------------------------------------------------------------
Failures:
<NO_OTHER_FAILURES>
------------------------------------------------------------
Root Cause (first observed failure):
[0]:
time : 2024-07-02_19:42:02
host : ip-26-0-171-21.ec2.internal
rank : 0 (local_rank: 0)
exitcode : 1 (pid: 2579548)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
============================================================
srun: error: ip-26-0-171-21: task 0: Exited with exit code 1
W0702 19:42:06.568000 140526730376960 torch/distributed/elastic/rendezvous/dynamic_rendezvous.py:1252] The node 'ip-26-0-173-202.ec2.internal_615186_0' has failed to send a keep-alive heartbeat to the rendezvous 'none' due to an error of type RendezvousConnectionError.
E0702 19:42:07.209000 140532397197120 torch/distributed/elastic/multiprocessing/api.py:826] failed (exitcode: 1) local_rank: 0 (pid: 615259) of binary: /fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/python3.10
W0702 19:42:07.215000 140532397197120 torch/distributed/elastic/rendezvous/dynamic_rendezvous.py:1203] The node 'ip-26-0-173-202.ec2.internal_615186_0' has failed to shutdown the rendezvous 'none' due to an error of type RendezvousConnectionError.
W0702 19:42:07.242000 140532397197120 torch/distributed/elastic/rendezvous/dynamic_rendezvous.py:1203] The node 'ip-26-0-173-202.ec2.internal_615186_0' has failed to shutdown the rendezvous 'none' due to an error of type RendezvousConnectionError.
W0702 19:42:07.273000 140532397197120 torch/distributed/elastic/rendezvous/dynamic_rendezvous.py:1203] The node 'ip-26-0-173-202.ec2.internal_615186_0' has failed to shutdown the rendezvous 'none' due to an error of type RendezvousConnectionError.
Traceback (most recent call last):
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/bin/torchrun", line 8, in <module>
sys.exit(main())
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/elastic/multiprocessing/errors/__init__.py", line 347, in wrapper
return f(*args, **kwargs)
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/run.py", line 879, in main
run(args)
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/run.py", line 870, in run
elastic_launch(
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 132, in __call__
return launch_agent(self._config, self._entrypoint, list(args))
File "/fsx/ferdinandmom/miniforge3/envs/env-bench-cluster/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 263, in launch_agent
raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError:
============================================================
/fsx/ferdinandmom/ferdinand-hf/bench_cluster/nanotron/run_train.py FAILED
------------------------------------------------------------
Failures:
[1]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 9 (local_rank: 1)
exitcode : 1 (pid: 615260)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
[2]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 10 (local_rank: 2)
exitcode : 1 (pid: 615261)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
[3]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 11 (local_rank: 3)
exitcode : 1 (pid: 615262)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
[4]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 12 (local_rank: 4)
exitcode : 1 (pid: 615263)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
[5]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 13 (local_rank: 5)
exitcode : 1 (pid: 615264)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
[6]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 14 (local_rank: 6)
exitcode : 1 (pid: 615265)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
[7]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 15 (local_rank: 7)
exitcode : 1 (pid: 615266)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
------------------------------------------------------------
Root Cause (first observed failure):
[0]:
time : 2024-07-02_19:42:07
host : ip-26-0-173-202.ec2.internal
rank : 8 (local_rank: 0)
exitcode : 1 (pid: 615259)
error_file: <N/A>
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
============================================================
srun: error: ip-26-0-173-202: task 1: Exited with exit code 1
Consider using `hf_transfer` for faster uploads. This solution comes with some limitations. See https://huggingface.co/docs/huggingface_hub/hf_transfer for more details.