Wrong function call in model_parallel_v2

[juna962]

Hello everyone,

There is a bug in the following Python file:
training/distributed_training/pytorch/model_parallel_v2/shared-scripts/logging_utils.py

At line 151, the code reads:

python
Copy code
avg_tflops = compute_tflops(avg_throughput, num_params, world_size, batch_seqlen)
However, the function definition in:
training/distributed_training/pytorch/model_parallel_v2/shared-scripts/train_utils.py
at line 36 is as follows:

python
Copy code
def compute_tflops(args, global_batch_size, step_time, world_size):
The arguments in the function call should be updated to match the definition. Specifically, args must be passed as the first argument. Alternatively, a new function needs to be defined to accommodate the current arguments.