|
@@ -65,10 +65,6 @@ ARG nvcc_threads=8
|
|
|
ENV NVCC_THREADS=${nvcc_threads}
|
|
|
|
|
|
ENV APHRODITE_INSTALL_PUNICA_KERNELS=1
|
|
|
-
|
|
|
-ENV APHRODITE_INSTALL_QUANT_KERNELS=1
|
|
|
-
|
|
|
-ENV APHRODITE_INSTALL_HADAMARD_KERNELS=1
|
|
|
|
|
|
ENV CCACHE_DIR=/root/.cache/ccache
|
|
|
RUN --mount=type=cache,target=/root/.cache/ccache \
|
|
@@ -91,7 +87,7 @@ RUN apt-get update -y \
|
|
|
|
|
|
RUN ldconfig /usr/local/cuda-12.4/compat/
|
|
|
|
|
|
-
|
|
|
+
|
|
|
RUN --mount=type=bind,from=build,src=/workspace/dist,target=/aphrodite-workspace/dist \
|
|
|
--mount=type=cache,target=/root/.cache/pip \
|
|
|
pip install dist/*.whl --verbose
|