Skip to content

Commit

Permalink
Exclude nvbandwidth with -o.
Browse files Browse the repository at this point in the history
  • Loading branch information
hongtaozhang committed Nov 20, 2024
1 parent 1f7c8e4 commit cf84e5d
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 4 deletions.
2 changes: 1 addition & 1 deletion dockerfile/cuda11.1.1.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -149,7 +149,7 @@ ADD dockerfile/etc /opt/microsoft/
WORKDIR ${SB_HOME}

ADD third_party third_party
RUN make -C third_party cuda
RUN make -C third_party cuda -o nvbandwidth

ADD . .
RUN python3 -m pip install --upgrade setuptools==65.7 && \
Expand Down
2 changes: 1 addition & 1 deletion dockerfile/cuda12.4.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ ADD dockerfile/etc /opt/microsoft/
WORKDIR ${SB_HOME}

ADD third_party third_party
RUN make -C third_party cuda_with_msccl_and_nvbandwidth
RUN make -C third_party cuda_with_msccl

ADD . .
RUN python3 -m pip install --upgrade setuptools==65.7 && \
Expand Down
3 changes: 1 addition & 2 deletions third_party/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,8 @@ NUM_MAKE_JOBS ?= $(shell nproc --ignore=2)

# Build targets.
all: cuda rocm
cuda_with_msccl_and_nvbandwidth: cuda cuda_msccl nvbandwidth
cuda_with_msccl: cuda cuda_msccl
cuda: common cuda_cutlass cuda_bandwidthTest cuda_nccl_tests cuda_perftest gpcnet cuda_gpuburn megatron_lm megatron_deepspeed
cuda: common cuda_cutlass cuda_bandwidthTest cuda_nccl_tests cuda_perftest gpcnet cuda_gpuburn megatron_lm megatron_deepspeed nvbandwidth
rocm: common rocm_perftest rocm_rccl_tests rocm_rocblas rocm_bandwidthTest rocm_hipblaslt megatron_deepspeed apex_rocm
cpu: common cpu_perftest cpu_stream
common: fio
Expand Down

0 comments on commit cf84e5d

Please sign in to comment.