已合并
修复部分st问题 #481
pengnuoheng创建于 2025年7月15日
修复部分st问题 #481
已合并
从refs/pull/481/head合入到master
共 4 个文件变更+4-1
| @@ -279,7 +279,7 @@ def metrics_sort(metrics, time_all) -> Dict[str, Tensor]: | |||
| 279 | return sorted_metric | 279 | return sorted_metric |
| 280 | 280 | ||
| 281 | 281 | ||
| 282 | -def compute_tps(compute_kwargs, metrics_result, gbs, n_samples, time_all, log_max_throughput): | 282 | +def compute_tps(compute_kwargs, metrics_result, gbs, n_samples, time_all, log_max_throughput=False): |
| 283 | actor_resource = compute_kwargs.get('actor_resource', {}) | 283 | actor_resource = compute_kwargs.get('actor_resource', {}) |
| 284 | reference_resource = compute_kwargs.get('reference_resource', {}) | 284 | reference_resource = compute_kwargs.get('reference_resource', {}) |
| 285 | reward_resource = compute_kwargs.get('reward_resource', None) | 285 | reward_resource = compute_kwargs.get('reward_resource', None) |
| @@ -1,4 +1,5 @@ | |||
| 1 | #!/bin/bash | 1 | #!/bin/bash |
| 2 | +set -e # 开启错误检测，当脚本中的任何命令返回非零退出状态时，立即退出脚本的执行。 | ||
| 2 | export CUDA_DEVICE_MAX_CONNECTIONS=1 | 3 | export CUDA_DEVICE_MAX_CONNECTIONS=1 |
| 3 | export HCCL_DETERMINISTIC=True | 4 | export HCCL_DETERMINISTIC=True |
| 4 | 5 | ||
| @@ -1,4 +1,5 @@ | |||
| 1 | #!/bin/bash | 1 | #!/bin/bash |
| 2 | +set -e # 开启错误检测，当脚本中的任何命令返回非零退出状态时，立即退出脚本的执行。 | ||
| 2 | export CUDA_DEVICE_MAX_CONNECTIONS=1 | 3 | export CUDA_DEVICE_MAX_CONNECTIONS=1 |
| 3 | export HCCL_DETERMINISTIC=True | 4 | export HCCL_DETERMINISTIC=True |
| 4 | 5 | ||
| @@ -1,4 +1,5 @@ | |||
| 1 | #!/bin/bash | 1 | #!/bin/bash |
| 2 | +set -e # 开启错误检测，当脚本中的任何命令返回非零退出状态时，立即退出脚本的执行。 | ||
| 2 | export CUDA_DEVICE_MAX_CONNECTIONS=1 | 3 | export CUDA_DEVICE_MAX_CONNECTIONS=1 |
| 3 | export HCCL_DETERMINISTIC=True | 4 | export HCCL_DETERMINISTIC=True |
| 4 | export HCCL_HOST_SOCKET_PORT_RANGE="60000-60050" | 5 | export HCCL_HOST_SOCKET_PORT_RANGE="60000-60050" |