Skip to content

GitLab

Unverified Commit 96bb8aa6 authored by rongfu.leng's avatar rongfu.leng Committed by GitHub
Browse files

[Bugfix] fix gpu docker image mis benchmarks dir (#16628) 


Signed-off-by: default avatarrongfu.leng <rongfu.leng@daocloud.io>
parent 3badb021
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
docker/Dockerfile
View file @ 96bb8aa6
...@@ -240,6 +240,7 @@ if [ "$TARGETPLATFORM" != "linux/arm64" ]; then \ ...@@ -240,6 +240,7 @@ if [ "$TARGETPLATFORM" != "linux/arm64" ]; then \
uv pip install --system https://github.com/flashinfer-ai/flashinfer/releases/download/v0.2.1.post2/flashinfer_python-0.2.1.post2+cu124torch2.6-cp38-abi3-linux_x86_64.whl ; \ uv pip install --system https://github.com/flashinfer-ai/flashinfer/releases/download/v0.2.1.post2/flashinfer_python-0.2.1.post2+cu124torch2.6-cp38-abi3-linux_x86_64.whl ; \
fi fi
COPY examples examples COPY examples examples
COPY benchmarks benchmarks
# Although we build Flashinfer with AOT mode, there's still # Although we build Flashinfer with AOT mode, there's still
# some issues w.r.t. JIT compilation. Therefore we need to # some issues w.r.t. JIT compilation. Therefore we need to
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment