cuda : rename build flag to LLAMA_CUDA (#6299)
This commit is contained in:
1 parent
b06c16ef9f
commit
280345968d
28 files changed
+129
-115
No files matched your search
@@ -20,8 +20,8 @@ COPY . .
|
||||
|
||||
# Set nvcc architecture
|
||||
ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
|
||||
# Enable cuBLAS
|
||||
ENV LLAMA_CUBLAS=1
|
||||
# Enable CUDA
|
||||
ENV LLAMA_CUDA=1
|
||||
|
||||
RUN make
|
||||
|
||||
|
||||
Reference in new issue
Block a user