From a0c70ca0995fd4bcb1643d717815294283d33698 Mon Sep 17 00:00:00 2001 From: Mohit Sharma Date: Fri, 13 Dec 2024 17:16:43 +0000 Subject: [PATCH] update moe-kernels commit --- Dockerfile_amd | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile_amd b/Dockerfile_amd index beb9101d..dc748f49 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -270,9 +270,9 @@ RUN python setup.py build FROM kernel-builder AS moe-kernels WORKDIR /usr/src -ENV MOE_KERNELS_BRANCH=b74b163a2e28042068ac8355e07e6dde926a967a +ENV MOE_KERNELS_BRANCH=a67b35841774b2056a73806c36661134b5054edd ENV VLLM_TARGET_DEVICE=rocm -RUN git clone https://github.com/mht-sharma/moe-kernels.git && \ +RUN git clone https://github.com/danieldk/moe-kernels.git && \ cd moe-kernels && \ git checkout ${MOE_KERNELS_BRANCH} && \ python setup.py install