Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/main' into upstream_merge_24_09_…
Browse files Browse the repository at this point in the history
…27_0.6.2
  • Loading branch information
gshtras committed Sep 27, 2024
2 parents 3d2bd9b + 956b831 commit 4f57e44
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Dockerfile.rocm
Original file line number Diff line number Diff line change
Expand Up @@ -248,10 +248,10 @@ RUN --mount=type=bind,from=export_pytorch,src=/,target=/install \
RUN python3 -m pip install --upgrade numba scipy huggingface-hub[cli]

RUN git clone -b nvtx_enabled https://github.com/ROCm/rocmProfileData.git \
&& cd rocmProfileData/rpd_tracer
&& cd rocmProfileData/rpd_tracer \
&& pip install -r requirements.txt && cd ../ \
&& make && make install \
cd hipMarker && python setup.py install
&& cd hipMarker && python setup.py install

# Install vLLM (and gradlib)
# Make sure punica kernels are built (for LoRA)
Expand Down

0 comments on commit 4f57e44

Please sign in to comment.