Skip to content

Commit

Permalink
Fix Dockerfile.rocm (#215)
Browse files Browse the repository at this point in the history
  • Loading branch information
gshtras authored Sep 27, 2024
1 parent 0a5881d commit 956b831
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 956b831

Please sign in to comment.