run-llama2.sh 952 B

123456789101112131415161718192021222324252627
  1. #!/usr/bin/env bash
  2. # MIT license
  3. # Copyright (C) 2024 Intel Corporation
  4. # SPDX-License-Identifier: MIT
  5. export ONEAPI_DEVICE_SELECTOR="level_zero:0"
  6. source /opt/intel/oneapi/setvars.sh
  7. #export GGML_SYCL_DEBUG=1
  8. #ZES_ENABLE_SYSMAN=1, Support to get free memory of GPU by sycl::aspect::ext_intel_free_memory. Recommended to use when --split-mode = layer.
  9. INPUT_PROMPT="Building a website can be done in 10 simple steps:\nStep 1:"
  10. MODEL_FILE=models/llama-2-7b.Q4_0.gguf
  11. NGL=99
  12. CONTEXT=4096
  13. if [ $# -gt 0 ]; then
  14. GGML_SYCL_DEVICE=$1
  15. echo "use $GGML_SYCL_DEVICE as main GPU"
  16. #use signle GPU only
  17. ZES_ENABLE_SYSMAN=1 ./build/bin/llama-cli -m ${MODEL_FILE} -p "${INPUT_PROMPT}" -n 400 -e -ngl ${NGL} -s 0 -c ${CONTEXT} -mg $GGML_SYCL_DEVICE -sm none
  18. else
  19. #use multiple GPUs with same max compute units
  20. ZES_ENABLE_SYSMAN=1 ./build/bin/llama-cli -m ${MODEL_FILE} -p "${INPUT_PROMPT}" -n 400 -e -ngl ${NGL} -s 0 -c ${CONTEXT}
  21. fi