run-llama2.sh 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. #!/bin/bash
  2. # MIT license
  3. # Copyright (C) 2024 Intel Corporation
  4. # SPDX-License-Identifier: MIT
  5. INPUT2="Building a website can be done in 10 simple steps:\nStep 1:"
  6. source /opt/intel/oneapi/setvars.sh
  7. if [ $# -gt 0 ]; then
  8. GGML_SYCL_DEVICE=$1
  9. GGML_SYCL_SINGLE_GPU=1
  10. else
  11. GGML_SYCL_DEVICE=0
  12. GGML_SYCL_SINGLE_GPU=0
  13. fi
  14. #export GGML_SYCL_DEBUG=1
  15. #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.
  16. if [ $GGML_SYCL_SINGLE_GPU -eq 1 ]; then
  17. echo "use $GGML_SYCL_DEVICE as main GPU"
  18. #use signle GPU only
  19. ZES_ENABLE_SYSMAN=1 ./build/bin/main -m models/llama-2-7b.Q4_0.gguf -p "${INPUT2}" -n 400 -e -ngl 33 -s 0 -mg $GGML_SYCL_DEVICE -sm none
  20. else
  21. #use multiple GPUs with same max compute units
  22. ZES_ENABLE_SYSMAN=1 ./build/bin/main -m models/llama-2-7b.Q4_0.gguf -p "${INPUT2}" -n 400 -e -ngl 33 -s 0
  23. fi
  24. #use main GPU only
  25. #ZES_ENABLE_SYSMAN=1 ./build/bin/main -m models/llama-2-7b.Q4_0.gguf -p "${INPUT2}" -n 400 -e -ngl 33 -s 0 -mg $GGML_SYCL_DEVICE -sm none
  26. #use multiple GPUs with same max compute units
  27. #ZES_ENABLE_SYSMAN=1 ./build/bin/main -m models/llama-2-7b.Q4_0.gguf -p "${INPUT2}" -n 400 -e -ngl 33 -s 0