Родитель
0d6e723c63
Коммит
63cf2f1db4
|
@ -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:
|
||||
|
|
Загрузка…
Ссылка в новой задаче