ML-gfx906/llama.cpp/env.sh at master · mixa3607/ML-gfx906 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#!/usr/bin/env bash

pushd $(dirname ${BASH_SOURCE[0]})

if [ "$LLAMA_IMAGE" == "" ]; then
  LLAMA_IMAGE=docker.io/mixa3607/llama.cpp-gfx906
fi

# rocm ver
if [ "$LLAMA_ROCM_VERSION" == "" ]; then
  LLAMA_ROCM_VERSION=7.14
fi

if [ "$LLAMA_REPO" == "" ]; then
  LLAMA_REPO="https://github.com/ggml-org/llama.cpp.git"
fi
if [ "$LLAMA_BRANCH" == "" ]; then
  LLAMA_BRANCH="master"
fi
if [ "$LLAMA_COMMIT" == "" ]; then
  LLAMA_COMMIT=""
fi
if [ "$LLAMA_CMAKE_HIP_FLAGS" == "" ]; then
  LLAMA_CMAKE_HIP_FLAGS=""
fi
if [ "$LLAMA_CCACHE_MAXSIZE" == "" ]; then
  LLAMA_CCACHE_MAXSIZE="2G"
fi
if [ "$LLAMA_IS_RELEASE" == "" ]; then
  LLAMA_IS_RELEASE="0"
fi
if [ "$LLAMA_PATCH" == "" ]; then
  LLAMA_PATCH="empty.patch"
fi

# push image
if [ "$LLAMA_PUSH" == "" ]; then
  LLAMA_PUSH="1"
fi

popd