fix relative path in flash-attn build:

2023-05-25 12:18:28 -04:00
parent 951facbb1f
commit 8d6a28953f
1 changed files with 3 additions and 3 deletions
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -43,11 +43,11 @@ RUN git clone https://github.com/HazyResearch/flash-attention.git && \
    python3 setup.py bdist_wheel && \
    cd csrc/fused_dense_lib && \
    python3 setup.py bdist_wheel && \
-    cd ../csrc/xentropy && \
+    cd ../xentropy && \
    python3 setup.py bdist_wheel && \
-    cd ../csrc/rotary && \
+    cd ../rotary && \
    python3 setup.py bdist_wheel && \
-    cd ../csrc/layer_norm && \
+    cd ../layer_norm && \
    python3 setup.py bdist_wheel

 FROM base-builder AS deepspeed-builder