Update.
This commit is contained in:
Yifan Xiong 2023-04-10 09:26:36 +08:00
Родитель 0d6e723c63
Коммит 63cf2f1db4
2 изменённых файлов: 8 добавлений и 1 удалений

Просмотреть файл

@ -128,3 +128,10 @@ RUN python3 -m pip install --no-cache-dir .[nvworker] && \
make cppbuild && \
make postinstall && \
rm -rf .git
# Update NCCL
RUN cd /usr/local && \
git clone -b v2.17.1-1 https://github.com/NVIDIA/nccl.git && \
cd nccl && \
make -j ${NUM_MAKE_JOBS} src.build && \
make install

Просмотреть файл

@ -182,7 +182,7 @@ def gen_traffic_pattern_host_groups(host_list, pattern, mpi_pattern_path, benchm
logger.error('Unsupported traffic pattern: {}'.format(pattern.type))
host_groups = __convert_config_to_host_group(config, host_list)
# write traffic pattern host groups to specified path
if pattern.mpi_pattern:
if pattern.type:
with open(mpi_pattern_path, 'a') as f:
f.write('benchmark_name: {} pattern_type: {}'.format(benchmark_name, pattern.type) + '\n')
for host_group in host_groups: