From f5fa3d131b33c1fedf963a0f6e8c4fa0aae8a10f Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Thu, 25 May 2023 11:29:15 -0400
Subject: [PATCH] fix cd within flash-attn

---
 docker/Dockerfile-base | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base
index 943bae3b0..5d84a7c24 100644
--- 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 ../csrc/xentropy && \
     python3 setup.py bdist_wheel && \
-    cd csrc/rotary && \
+    cd ../csrc/rotary && \
     python3 setup.py bdist_wheel && \
-    cd csrc/layer_norm && \
+    cd ../csrc/layer_norm && \
     python3 setup.py bdist_wheel
 
 FROM base-builder AS deepspeed-builder