1. 08 Dec, 2025 1 commit
  2. 18 Nov, 2025 1 commit
  3. 12 Nov, 2025 1 commit
  4. 04 Nov, 2025 1 commit
  5. 14 Oct, 2025 1 commit
    • Thomas Stocker's avatar
      Vulkan based on #9650 (#11835) · 2aba569a
      Thomas Stocker authored
      * implement the vulkan C backend
      
      * add support in gpu.go
      
      * add support in gen_linux.sh
      
      * it builds
      
      * fix segfault
      
      * fix compilation
      
      * fix free memory monitor
      
      * fix total memory monitor
      
      * update gpu.go
      
      * fix build
      
      * fix check_perfmon len
      
      * remove cap_get_bound check
      
      * fix vulkan handle releasing
      
      * fix build on federa 40
      
      * fix vulkan on windows
      
      * making amdgpu work on arm achitecutre with vulkan
      
      * add x86_64 lines in VulkanGlobs and capLinuxGlobs
      
      * add aarch64 lines in vulkanGlobs and capLinuxGlobs
      
      * Fix variable name
      
      * Add vulkan build patch from @jmorganca
      
      * Sync vendored ggml to add Vulkan support
      
      * Updated dockerfile
      
      https://github.com/whyvl/ollama-vulkan/issues/7#issuecomment-2660836871
      
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Installing rocm library
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * This version works well
      
      built based on this: https://github.com/whyvl/ollama-vulkan/issues/7#issuecomment-2660836871
      
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Applied 00-fix-vulkan-building.patch
      
      Work done by McBane87 here: https://github.com/whyvl/ollama-vulkan/issues/7#issuecomment-2660836871
      
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Fixed the "detached head" issues
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Merged in the right direction
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Merging the latest stable (#2)
      
      * Applied 00-fix-vulkan-building.patch
      
      * Implemented vulkan backend based on the work done by whyvl, Dts0, McBane87 and others
      
      Tested on AMD Ryzen 7 8845HS w/ Radeon 780M Graphics with ROCm disabled
      
      ```
      [GIN-debug] POST   /v1/chat/completions      --> github.com/ollama/ollama/server.(*Server).ChatHandler-fm (6 handlers)
      [GIN-debug] POST   /v1/completions           --> github.com/ollama/ollama/server.(*Server).GenerateHandler-fm (6 handlers)
      [GIN-debug] POST   /v1/embeddings            --> github.com/ollama/ollama/server.(*Server).EmbedHandler-fm (6 handlers)
      [GIN-debug] GET    /v1/models                --> github.com/ollama/ollama/server.(*Server).ListHandler-fm (6 handlers)
      [GIN-debug] GET    /v1/models/:model         --> github.com/ollama/ollama/server.(*Server).ShowHandler-fm (6 handlers)
      time=2025-03-11T13:00:40.793Z level=INFO source=gpu.go:199 msg="vulkan: load libvulkan and libcap ok"
      time=2025-03-11T13:00:40.877Z level=INFO source=gpu.go:421 msg="error looking up vulkan GPU memory" error="device is a CPU"
      time=2025-03-11T13:00:40.878Z level=WARN source=amd_linux.go:443 msg="amdgpu detected, but no compatible rocm library found.  Either install rocm v6, or follow manual install instructions at https://github.com/ollama/ollama/blob/main/docs/linux.md#manual-install"
      time=2025-03-11T13:00:40.878Z level=WARN source=amd_linux.go:348 msg="unable to verify rocm library: no suitable rocm found, falling back to CPU"
      time=2025-03-11T13:00:40.879Z level=INFO source=types.go:137 msg="inference compute" id=0 library=vulkan variant="" compute=1.3 driver=1.3 name="AMD Radeon Graphics (RADV GFX1103_R1)" total="15.6 GiB" available="15.6 GiB"
      ```
      
      ```
       # ollama run phi4:14b
      >>> /set verbose
      Set 'verbose' mode.
      >>> how's it going?
      Hello! I'm here to help you with any questions or tasks you have. How can I assist you today? 😊
      
      
      
      total duration:       3.341959745s
      load duration:        18.165612ms
      prompt eval count:    15 token(s)
      prompt eval duration: 475ms
      prompt eval rate:     31.58 tokens/s
      eval count:           26 token(s)
      eval duration:        2.846s
      eval rate:            9.14 tokens/s
      >>>
      ```
      
      * This is no longer needed
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Fixes SIGSEGV: segmentation violation running gemma3 models on ollama 0.6.0 #21
      
      Patch provided by McBane87 on https://github.com/whyvl/ollama-vulkan/issues/21
      
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Applied 04-disable-mmap-vulkan.patch
      
      From: https://github.com/whyvl/ollama-vulkan/issues/7#issuecomment-2660836871
      
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Pulled new upstream code for ggml-bulkan backend
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Merged latest ollama 0.6.2 and nasrally's Flash Attention patches (#5)
      
      * readme: add Ellama to list of community integrations (#9800)
      
      * readme: add screenpipe to community integrations (#9786)
      
      * Add support for ROCm gfx1151 (#9773)
      
      * conditionally enable parallel pipelines
      
      * sample: make mutations in transforms explicit (#9743)
      
      * updated minP to use early exit making use of sorted tokens
      
      * ml/backend/ggml: allocate memory with malloc when loading model (#9822)
      
      * runner: remove cache prompt flag from ollama runner (#9826)
      
      We do not need to bypass the prompt caching in the ollama runner yet, as
      only embedding models needed to bypass the prompt caching. When embedding
      models are implemented they can skip initializing this cache completely.
      
      * ollamarunner: Check for minBatch of context space when shifting
      
      Models can specify that a group of inputs need to be handled a single
      batch. However, context shifting didn't respect this and could trigger
      a break anyways. In this case, we should instead trigger a context
      shift earlier so that it occurs before the grouped batch.
      
      Note that there still some corner cases:
       - A long prompt that exceeds the context window can get truncated
         in the middle of an image. With the current models, this will
         result in the model not recognizing the image at all, which is
         pretty much the expected result with truncation.
       - The context window is set less than the minimum batch size. The
         only solution to this is to refuse to load the model with these
         settings. However, this can never occur with current models and
         default settings.
      
      Since users are unlikely to run into these scenarios, fixing them is
      left as a follow up.
      
      * Applied latest patches from McBane87
      
      See this for details: https://github.com/whyvl/ollama-vulkan/issues/7#issuecomment-2708820861
      
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      
      * Add ability to enable flash attention on vulkan (#4
      
      )
      
      * discover: add flash attention handling for vulkan
      * envconfig: fix typo in config.go
      
      As part of the process some code was refactored and I added a new field
      FlashAttention to GpuInfo since the previous solution didn't allow for a
      granular check via vulkan extensions. As a side effect, this now allows
      for granular per-device FA support checking in other places
      
      ---------
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      Co-authored-by: default avatarzeo <108888572+zeozeozeo@users.noreply.github.com>
      Co-authored-by: default avatarLouis Beaumont <louis.beaumont@gmail.com>
      Co-authored-by: default avatarDaniel Hiltgen <dhiltgen@users.noreply.github.com>
      Co-authored-by: default avatarMichael Yang <mxyng@pm.me>
      Co-authored-by: default avatarParth Sareen <parth.sareen@ollama.com>
      Co-authored-by: default avatarJeffrey Morgan <jmorganca@gmail.com>
      Co-authored-by: default avatarBruce MacDonald <brucewmacdonald@gmail.com>
      Co-authored-by: default avatarJesse Gross <jesse@ollama.com>
      Co-authored-by: default avatarNikita <50599445+nasrally@users.noreply.github.com>
      
      * Revert Readme changes
      
      * Revert
      
      * Revert changes in amd_linux.go
      
      * Revert changes in amd_linux.go
      
      * Remove flashattention setting gpu.go
      
      * Revert whitespace changes in gpu.go
      
      * Revert changes in transforms_test.go
      
      * Revert changes in runner.go
      
      * Revert changes in Makefile.sync
      
      * Revert some unintented changes in Dockerfile
      
      * Revert vulkan copy changes in Dockerfile
      
      * Update Vulkan Code to de4c07f93783a1a96456a44dc16b9db538ee1618
      
      * Fixed duplicate sync in ggml.go
      
      * Revert changes in ggml.go
      
      * Revert chnages in ggml.go
      
      * enable falsh attention on vulkan
      
      * revert remove parenthesis
      
      * fixed flash attention logic enabling
      
      * vk_check_flash_attention 0 means supported
      
      * Update gpu.go
      
      * Add vulkan to Windows Build script
      
      * Remove commented out code
      
      * Enable Vulkan Flash attention in FlashAttentionSupported
      
      * Fix logging
      
      * Update Vulkan backend to e54d41befcc1575f4c898c5ff4ef43970cead75f
      
      * Removed libcap related code
      
      libcap is not directly related to Vulkan and should be added by its own PR. It adds additional library dependencies for building and also requires users to run setcap or run ollama as root, which is not ideal for easy use
      
      * Fix Unit Test (Add Vulkan Library)
      
      * Add vulkan to TestHomogeneousGPUs
      Test
      
      * vulkan: get GPU ID (ollama v0.11.5)
      Signed-off-by: default avatarXiaodong Ye <xiaodong.ye@mthreads.com>
      
      * disable mmap for vulkan
      
      * Reduce Changes remove TestHomogeneousGPUs (doesn't exist on master)
      
      * Update vulkan version to the version used in llama.cpp
      
      * rename gpu patch to correct number
      
      * added Vulkan API to get correct Device UUID
      
      current UUID from pipelineCacheUUID does not match CUDA
      
      * Fix GPU ID Patch
      
      * Remove Code not in llama.cpp
      
      * modified UUID code inside ggml
      
      * Fix Patch
      
      * Copied minimal definition from vulkan header
      
      * Fix compile error in Mac
      
      Metal is preferred so we're disabling Vulkan for now
      
      * Removed unused code
      
      Fix linter error in CI
      
      * Fix patches apply
      
      * fixing lint error
      
      * Removed unneeded function call
      
      Somehow removing this call fixed the crashing when Vulkan header was removed
      
      * added missing NL
      
      * Fixed missing members in Vulkan header
      
      also added zero clear for some structs
      
      * Fixed wrong structure ID
      
      * Fixed Vulkan header
      
      More aligned with official header definition now
      
      * buildvulkanAsSeperateFunction
      
      * Vulkan on Windows Test
      
      * temporarly comment out gate to run windows task
      
      * use temporarly windows-latest for build
      
      * Commenting out other presets to build vulkan
      
      * reenable cpu
      
      * commenting out error action stop
      
      * temporarly commenting out rocm
      
      * set vulkan path
      
      * comment out cude for faster turnaround
      
      * correct vulkan install
      
      * correct vulkan silent install
      
      * fixed install command
      
      * revert debugging changes (vulkan builds on windows)
      
      * revert windows-latest
      
      * trying to build vulkan for linux
      
      * temporarly disable cuda and rocm
      
      * try again linux build
      
      * fix version
      
      * trying to fix
      
      * trying again
      
      * trying again
      
      * fix version
      
      * fixed vulkan-sdk name
      
      * try again
      
      * trying again
      
      * try without version number
      
      * try again
      
      * add some more extra
      
      * trying to use version 1.4.313
      
      * revert debugging changes
      
      * Filter out already supported gpus
      
      * revert debug code
      
      * Use runners for GPU discovery
      
      This revamps how we discover GPUs in the system by leveraging the Ollama
      runner.  This should eliminate inconsistency between our GPU discovery and the
      runners capabilities at runtime, particularly for cases where we try to filter
      out unsupported GPUs.  Now the runner does that implicitly based on the actual
      device list.  In some cases free VRAM reporting can be unreliable which can
      leaad to scheduling mistakes, so this also includes a patch to leverage more
      reliable VRAM reporting libraries if available.
      
      Automatic workarounds have been removed as only one GPU leveraged this, which
      is now documented. This GPU will soon fall off the support matrix with the next
      ROCm bump.
      
      Additional cleanup of the scheduler and discovery packages can be done in the
      future once we have switched on the new memory management code, and removed
      support for the llama runner.
      
      * timing info for runner
      
      * WIP - wire up Vulkan with the new engine based discovery
      
      Not a complete implementation - free VRAM is better, but not accurate on
      windows
      
      * fix - trust the library paths from discovery when starting runner
      
      * fix index bug
      
      * fix vulkan ids to be underlying
      
      * fix - give bootstrapping more time on slow systems
      
      * Test if Vulkan device is supported
      
      * vk_check_flash_attention is not needed (coompat2 coopmapt and scalar implementation exist)
      
      * Handle GGML_VK_VISIBLE_DEVICES
      
      * ask for supported first
      
      * win: fix CPU query buffer handling
      
      Try in a short loop until we get the size right.
      
      * test: harden integration tests for slow start
      
      If the server takes a while to start up, block
      tests from starting until it's online to avoid
      setting large timeouts in individual test cases.
      
      * gofumpt fix
      
      * fix build
      
      * merge fixes
      
      * merge fixes
      
      * fixed build
      
      * merge fixes
      
      * fixing build
      
      * fixed build
      
      * fixed formatting
      
      * fixed build
      
      * fix vulkan gpu id patch
      
      * sync llama.cpp vulkan code
      
      * update build windows script
      
      * merge fixes
      
      * fix format
      
      * fixed vulkan casing
      
      * handle igpu as gpu
      
      * improve case
      
      * print out unknown library
      
      * rturn Vulkan for vulkan library
      
      * Revert "rturn Vulkan for vulkan library"
      
      This reverts commit 690461a12fd5e93295d174c97edefb2bc33285b1.
      
      * fixed patch number
      
      * return Library Name
      
      * remvoe debug code
      
      * return integrated in vulkan backend
      
      * Return pci Properties
      
      * update patch
      
      * directly get pci proeprties without parsing
      
      * workaround for filtering devices. Correct way is to have a LibraryPosition Parameter in the deviceInfo
      
      * Revert "directly get pci proeprties without parsing"
      
      This reverts commit 8e0624851f5ed7d9f74518f574dfb422e4dd4dc2.
      
      * Set FilteredID for Environment Filtering
      
      * ROCm Library is named ROCm
      
      * revert changes in patch
      
      * Create 0028-vulkan-pci-and-memory.patch
      
      * vulkan memory patch
      
      * casing fix
      
      * Add more pci properties
      
      * Added better memory management
      
      * Added better memory managament
      
      * fixed patch
      
      * Fixed patch
      
      * FilterID creation group by library
      
      * filter out vulkan supported by other gpu
      
      * fixing deviceid compare
      
      * Vulkan Fix FA coopmat1 invalid array indexing
      
      * Use everywhere the same Vulkan Version 1.4.321.1
      
      * Remove unneeded patch
      
      * vulkan update
      
      * sync vulkan glsl files
      
      * only use for vulkan the filteredid (numeric device number)
      
      * simplify code
      
      ---------
      Signed-off-by: default avatarVadim Grinco <vadim@grinco.eu>
      Signed-off-by: default avatarXiaodong Ye <xiaodong.ye@mthreads.com>
      Co-authored-by: default avatarpufferffish <github@bandersnatch.anonaddy.com>
      Co-authored-by: KOISHI KOMEIJI FROM TOUHOU 11 <fuck>
      Co-authored-by: default avatarDSLstandard <qgeneral35@gmail.com>
      Co-authored-by: default avatarpufferffish <me@windtfw.com>
      Co-authored-by: default avataryeongbba <yeongmo.lee@logpresso.com>
      Co-authored-by: default avatartomaThomas <tomathomas@mailbox.org>
      Co-authored-by: default avatarAntoine Viallon <antoine@lesviallon.fr>
      Co-authored-by: default avatarVadim Grinco <vadim@grinco.eu>
      Co-authored-by: default avatarzeo <108888572+zeozeozeo@users.noreply.github.com>
      Co-authored-by: default avatarLouis Beaumont <louis.beaumont@gmail.com>
      Co-authored-by: default avatarDaniel Hiltgen <dhiltgen@users.noreply.github.com>
      Co-authored-by: default avatarMichael Yang <mxyng@pm.me>
      Co-authored-by: default avatarParth Sareen <parth.sareen@ollama.com>
      Co-authored-by: default avatarJeffrey Morgan <jmorganca@gmail.com>
      Co-authored-by: default avatarBruce MacDonald <brucewmacdonald@gmail.com>
      Co-authored-by: default avatarJesse Gross <jesse@ollama.com>
      Co-authored-by: default avatarNikita <50599445+nasrally@users.noreply.github.com>
      Co-authored-by: default avatarMasato Nakasaka <masato.nakasaka@intel.com>
      Co-authored-by: default avatarXiaodong Ye <xiaodong.ye@mthreads.com>
      Co-authored-by: default avatarDaniel Hiltgen <daniel@ollama.com>
      2aba569a
  6. 11 Sep, 2025 1 commit
  7. 10 Sep, 2025 1 commit
    • Daniel Hiltgen's avatar
      Add v12 + v13 cuda support (#12000) · 17a023f3
      Daniel Hiltgen authored
      * Add support for upcoming NVIDIA Jetsons
      
      The latest Jetsons with JetPack 7 are moving to an SBSA compatible model and
      will not require building a JetPack specific variant.
      
      * cuda: bring back dual versions
      
      This adds back dual CUDA versions for our releases,
      with v11 and v13 to cover a broad set of GPUs and
      driver versions.
      
      * win: break up native builds in build_windows.ps1
      
      * v11 build working on windows and linux
      
      * switch to cuda v12.8 not JIT
      
      * Set CUDA compression to size
      
      * enhance manual install linux docs
      17a023f3
  8. 24 Jun, 2025 2 commits
  9. 23 Jun, 2025 1 commit
    • Daniel Hiltgen's avatar
      Re-remove cuda v11 (#10694) · 1c6669e6
      Daniel Hiltgen authored
      * Re-remove cuda v11
      
      Revert the revert - drop v11 support requiring drivers newer than Feb 23
      
      This reverts commit c6bcdc42.
      
      * Simplify layout
      
      With only one version of the GPU libraries, we can simplify things down somewhat.  (Jetsons still require special handling)
      
      * distinct sbsa variant for linux arm64
      
      This avoids accidentally trying to load the sbsa cuda libraries on
      a jetson system which results in crashes.
      
      * temporary prevent rocm+cuda mixed loading
      1c6669e6
  10. 13 May, 2025 1 commit
  11. 07 May, 2025 1 commit
    • Daniel Hiltgen's avatar
      remove cuda v11 (#10569) · fa393554
      Daniel Hiltgen authored
      This reduces the size of our Windows installer payloads by ~256M by dropping
      support for nvidia drivers older than Feb 2023.  Hardware support is unchanged.
      
      Linux default bundle sizes are reduced by ~600M to 1G.
      fa393554
  12. 16 Apr, 2025 1 commit
  13. 27 Feb, 2025 2 commits
  14. 25 Feb, 2025 3 commits
    • Blake Mizerany's avatar
      .github: always run tests, and other helpful fixes (#9348) · 0d694793
      Blake Mizerany authored
      During work on our new registry client, I ran into frustrations with CI
      where a misspelling in a comment caused the linter to fail, which caused
      the tests to not run, which caused the build to not be cached, which
      caused the next run to be slow, which caused me to be sad.
      
      This commit address these issues, and pulls in some helpful changes
      we've had in CI on ollama.com for some time now.
      
      They are:
      
      * Always run tests, even if the other checks fail.
      
      Tests are the most important part of CI, and should always run. Failures
      in tests can be correlated with failures in other checks, and can help
      surface the root cause of the failure sooner. This is especially
      important when the failure is platform specific, and the tests are not
      platform independent.
      
      * Check that `go generate` is clean.
      
      This prevents 'go generate' abuse regressions. This codebase used to use
      it to generate platform specific binary build artifacts. Let's make sure
      that does not happen again and this powerful tool is used correctly, and
      the generated code is checked in.
      
      Also, while adding `go generate` the check, it was revealed that the
      generated metal code was putting dates in the comments, resulting in
      non-deterministic builds. This is a bad practice, and this commit fixes
      that. Git tells us the most important date: the commit date along with
      other associated changes.
      
      * Check that `go mod tidy` is clean.
      
      A new job to check that `go mod tidy` is clean was added, to prevent
      easily preventable merge conflicts or go.mod changes being deferred to a
      future PR that is unrelated to the change that caused the go.mod to
      change.
      
      * More robust caching.
      
      We now cache the go build cache, and the go mod download cache
      independently. This is because the download cache contains zips that can
      be unpacked in parallel faster than they can be fetched and extracted by
      tar. This speeds up the build significantly.
      
      The linter is hostile enough. It does not need to also punish us with
      longer build times due to small failures like misspellings.
      0d694793
    • Daniel Hiltgen's avatar
      Update ROCm (6.3 linux, 6.2 windows) and CUDA v12.8 (#9304) · e91ae3d4
      Daniel Hiltgen authored
      * Bump cuda and rocm versions
      
      Update ROCm to linux:6.3 win:6.2 and CUDA v12 to 12.8.
      Yum has some silent failure modes, so largely switch to dnf.
      
      * Fix windows build script
      e91ae3d4
    • Blake Mizerany's avatar
      server/internal: copy bmizerany/ollama-go to internal package (#9294) · 348b3e09
      Blake Mizerany authored
      This commit copies (without history) the bmizerany/ollama-go repository
      with the intention of integrating it into the ollama as a replacement
      for the pushing, and pulling of models, and management of the cache they
      are pushed and pulled from.
      
      New homes for these packages will be determined as they are integrated
      and we have a better understanding of proper package boundaries.
      348b3e09
  15. 06 Feb, 2025 1 commit
  16. 29 Jan, 2025 1 commit
    • Michael Yang's avatar
      next build (#8539) · dcfb7a10
      Michael Yang authored
      
      
      * add build to .dockerignore
      
      * test: only build one arch
      
      * add build to .gitignore
      
      * fix ccache path
      
      * filter amdgpu targets
      
      * only filter if autodetecting
      
      * Don't clobber gpu list for default runner
      
      This ensures the GPU specific environment variables are set properly
      
      * explicitly set CXX compiler for HIP
      
      * Update build_windows.ps1
      
      This isn't complete, but is close.  Dependencies are missing, and it only builds the "default" preset.
      
      * build: add ollama subdir
      
      * add .git to .dockerignore
      
      * docs: update development.md
      
      * update build_darwin.sh
      
      * remove unused scripts
      
      * llm: add cwd and build/lib/ollama to library paths
      
      * default DYLD_LIBRARY_PATH to LD_LIBRARY_PATH in runner on macOS
      
      * add additional cmake output vars for msvc
      
      * interim edits to make server detection logic work with dll directories like lib/ollama/cuda_v12
      
      * remove unncessary filepath.Dir, cleanup
      
      * add hardware-specific directory to path
      
      * use absolute server path
      
      * build: linux arm
      
      * cmake install targets
      
      * remove unused files
      
      * ml: visit each library path once
      
      * build: skip cpu variants on arm
      
      * build: install cpu targets
      
      * build: fix workflow
      
      * shorter names
      
      * fix rocblas install
      
      * docs: clean up development.md
      
      * consistent build dir removal in development.md
      
      * silence -Wimplicit-function-declaration build warnings in ggml-cpu
      
      * update readme
      
      * update development readme
      
      * llm: update library lookup logic now that there is one runner (#8587)
      
      * tweak development.md
      
      * update docs
      
      * add windows cuda/rocm tests
      
      ---------
      Co-authored-by: default avatarjmorganca <jmorganca@gmail.com>
      Co-authored-by: default avatarDaniel Hiltgen <daniel@ollama.com>
      dcfb7a10
  17. 11 Dec, 2024 1 commit
  18. 05 Dec, 2024 1 commit
  19. 12 Nov, 2024 1 commit
  20. 02 Nov, 2024 1 commit
  21. 30 Oct, 2024 2 commits
  22. 17 Oct, 2024 1 commit
  23. 08 Oct, 2024 1 commit
    • Jeffrey Morgan's avatar
      Re-introduce the `llama` package (#5034) · 96efd905
      Jeffrey Morgan authored
      
      
      * Re-introduce the llama package
      
      This PR brings back the llama package, making it possible to call llama.cpp and
      ggml APIs from Go directly via CGo. This has a few advantages:
      
      - C APIs can be called directly from Go without needing to use the previous
        "server" REST API
      - On macOS and for CPU builds on Linux and Windows, Ollama can be built without
        a go generate ./... step, making it easy to get up and running to hack on
        parts of Ollama that don't require fast inference
      - Faster build times for AVX,AVX2,CUDA and ROCM (a full build of all runners
        takes <5 min on a fast CPU)
      - No git submodule making it easier to clone and build from source
      
      This is a big PR, but much of it is vendor code except for:
      
      - llama.go CGo bindings
      - example/: a simple example of running inference
      - runner/: a subprocess server designed to replace the llm/ext_server package
      - Makefile an as minimal as possible Makefile to build the runner package for
        different targets (cpu, avx, avx2, cuda, rocm)
      Co-authored-by: default avatarJesse Gross <jesse@ollama.com>
      Co-authored-by: default avatarDaniel Hiltgen <daniel@ollama.com>
      
      * cache: Clear old KV cache entries when evicting a slot
      
      When forking a cache entry, if no empty slots are available we
      evict the least recently used one and copy over the KV entries
      from the closest match. However, this copy does not overwrite
      existing values but only adds new ones. Therefore, we need to
      clear the old slot first.
      
      This change fixes two issues:
       - The KV cache fills up and runs out of space even though we think
         we are managing it correctly
       - Performance gets worse over time as we use new cache entries that
         are not hot in the processor caches
      
      * doc: explain golang objc linker warning (#6830)
      
      * llama: gather transitive dependencies for rocm for dist packaging (#6848)
      
      * Refine go server makefiles to be more DRY (#6924)
      
      This breaks up the monolithic Makefile for the Go based runners into a
      set of utility files as well as recursive Makefiles for the runners.
      Files starting with the name "Makefile" are buildable, while files that
      end with ".make" are utilities to include in other Makefiles.  This
      reduces the amount of nearly identical targets and helps set a pattern
      for future community contributions for new GPU runner architectures.
      
      When we are ready to switch over to the Go runners, these files should
      move to the top of the repo, and we should add targets for the main CLI,
      as well as a helper "install" (put all the built binaries on the local
      system in a runnable state) and "dist" target (generate the various
      tar/zip files for distribution) for local developer use.
      
      * llama: don't create extraneous directories (#6988)
      
      * llama: Exercise the new build in CI (#6989)
      
      Wire up some basic sanity testing in CI for the Go runner.  GPU runners are not covered yet.
      
      * llama: Refine developer docs for Go server (#6842)
      
      This enhances the documentation for development focusing on the new Go
      server.  After we complete the transition further doc refinements
      can remove the "transition" discussion.
      
      * runner.go: Allocate batches for all sequences during init
      
      We should tell the model that we could have full batches for all
      sequences. We already do this when we allocate the batches but it was
      missed during initialization.
      
      * llama.go: Don't return nil from Tokenize on zero length input
      
      Potentially receiving nil in a non-error condition is surprising to
      most callers - it's better to return an empty slice.
      
      * runner.go: Remove stop tokens from cache
      
      If the last token is EOG then we don't return this and it isn't
      present in the cache (because it was never submitted to Decode).
      This works well for extending the cache entry with a new sequence.
      
      However, for multi-token stop sequences, we won't return any of the
      tokens but all but the last one will be in the cache. This means
      when the conversation continues the cache will contain tokens that
      don't overlap with the new prompt.
      
      This works (we will pick up the portion where there is overlap) but
      it causes unnecessary cache thrashing because we will fork the original
      cache entry as it is not a perfect match.
      
      By trimming the cache to the tokens that we actually return this
      issue can be avoided.
      
      * runner.go: Simplify flushing of pending tokens
      
      * runner.go: Update TODOs
      
      * runner.go: Don't panic when processing sequences
      
      If there is an error processing a sequence, we should return a
      clean HTTP error back to Ollama rather than panicing. This will
      make us more resilient to transient failures.
      
      Panics can still occur during startup as there is no way to serve
      requests if that fails.
      Co-authored-by: default avatarjmorganca <jmorganca@gmail.com>
      
      * runner.go: More accurately capture timings
      
      Currently prompt processing time doesn't capture the that it takes
      to tokenize the input, only decoding time. We should capture the
      full process to more accurately reflect reality. This is especially
      true once we start processing images where the initial processing
      can take significant time. This is also more consistent with the
      existing C++ runner.
      
      * runner.go: Support for vision models
      
      In addition to bringing feature parity with the C++ runner, this also
      incorporates several improvements:
       - Cache prompting works with images, avoiding the need to re-decode
         embeddings for every message in a conversation
       - Parallelism is supported, avoiding the need to restrict to one
         sequence at a time. (Though for now Ollama will not schedule
         them while we might need to fall back to the old runner.)
      Co-authored-by: default avatarjmorganca <jmorganca@gmail.com>
      
      * runner.go: Move Unicode checking code and add tests
      
      * runner.go: Export external cache members
      
      Runner and cache are in the same package so the change doesn't
      affect anything but it is more internally consistent.
      
      * runner.go: Image embedding cache
      
      Generating embeddings from images can take significant time (on
      my machine between 100ms and 8s depending on the model). Although
      we already cache the result of decoding these images, the embeddings
      need to be regenerated every time. This is not necessary if we get
      the same image over and over again, for example, during a conversation.
      
      This currently uses a very small cache with a very simple algorithm
      but it is easy to improve as is warranted.
      
      * llama: catch up on patches
      
      Carry forward solar-pro and cli-unicode patches
      
      * runner.go: Don't re-allocate memory for every batch
      
      We can reuse memory allocated from batch to batch since batch
      size is fixed. This both saves the cost of reallocation as well
      keeps the cache lines hot.
      
      This results in a roughly 1% performance improvement for token
      generation with Nvidia GPUs on Linux.
      
      * runner.go: Default to classic input cache policy
      
      The input cache as part of the go runner implemented a cache
      policy that aims to maximize hit rate in both single and multi-
      user scenarios. When there is a cache hit, the response is
      very fast.
      
      However, performance is actually slower when there is an input
      cache miss due to worse GPU VRAM locality. This means that
      performance is generally better overall for multi-user scenarios
      (better input cache hit rate, locality was relatively poor already).
      But worse for single users (input cache hit rate is about the same,
      locality is now worse).
      
      This defaults the policy back to the old one to avoid a regression
      but keeps the new one available through an environment variable
      OLLAMA_MULTIUSER_CACHE. This is left undocumented as the goal is
      to improve this in the future to get the best of both worlds
      without user configuration.
      
      For inputs that result in cache misses, on Nvidia/Linux this
      change improves performance by 31% for prompt processing and
      13% for token generation.
      
      * runner.go: Increase size of response channel
      
      Generally the CPU can easily keep up with handling reponses that
      are generated but there's no reason not to let generation continue
      and handle things in larger batches if needed.
      
      * llama: Add CI to verify all vendored changes have patches (#7066)
      
      Make sure we don't accidentally merge changes in the vendored code
      that aren't also reflected in the patches.
      
      * llama: adjust clip patch for mingw utf-16 (#7065)
      
      * llama: adjust clip patch for mingw utf-16
      
      * llama: ensure static linking of runtime libs
      
      Avoid runtime dependencies on non-standard libraries
      
      * runner.go: Enable llamafile (all platforms) and BLAS (Mac OS)
      
      These are two features that are shown on llama.cpp's system info
      that are currently different between the two runners. On my test
      systems the performance difference is very small to negligible
      but it is probably still good to equalize the features.
      
      * llm: Don't add BOS/EOS for tokenize requests
      
      This is consistent with what server.cpp currently does. It affects
      things like token processing counts for embedding requests.
      
      * runner.go: Don't cache prompts for embeddings
      
      Our integration with server.cpp implicitly disables prompt caching
      because it is not part of the JSON object being parsed, this makes
      the Go runner behavior similarly.
      
      Prompt caching has been seen to affect the results of text completions
      on certain hardware. The results are not wrong either way but they
      are non-deterministic. However, embeddings seem to be affected even
      on hardware that does not show this behavior for completions. For
      now, it is best to maintain consistency with the existing behavior.
      
      * runner.go: Adjust debug log levels
      
      Add system info printed at startup and quiet down noisier logging.
      
      * llama: fix compiler flag differences (#7082)
      
      Adjust the flags for the new Go server to more closely match the
      generate flow
      
      * llama: refine developer docs (#7121)
      
      * llama: doc and example clean up (#7122)
      
      * llama: doc and example clean up
      
      * llama: Move new dockerfile into llama dir
      
      Temporary home until we fully transition to the Go server
      
      * llama: runner doc cleanup
      
      * llama.go: Add description for Tokenize error case
      
      ---------
      Co-authored-by: default avatarJesse Gross <jesse@ollama.com>
      Co-authored-by: default avatarDaniel Hiltgen <daniel@ollama.com>
      Co-authored-by: default avatarDaniel Hiltgen <dhiltgen@users.noreply.github.com>
      96efd905
  24. 12 Sep, 2024 1 commit
    • Daniel Hiltgen's avatar
      Optimize container images for startup (#6547) · cd5c8f64
      Daniel Hiltgen authored
      * Optimize container images for startup
      
      This change adjusts how to handle runner payloads to support
      container builds where we keep them extracted in the filesystem.
      This makes it easier to optimize the cpu/cuda vs cpu/rocm images for
      size, and should result in faster startup times for container images.
      
      * Refactor payload logic and add buildx support for faster builds
      
      * Move payloads around
      
      * Review comments
      
      * Converge to buildx based helper scripts
      
      * Use docker buildx action for release
      cd5c8f64
  25. 13 Aug, 2024 1 commit
  26. 02 Aug, 2024 1 commit
  27. 22 Jul, 2024 1 commit
  28. 15 Jul, 2024 1 commit
  29. 10 Jul, 2024 1 commit
    • Daniel Hiltgen's avatar
      Bump ROCm on windows to 6.1.2 · 1f50356e
      Daniel Hiltgen authored
      This also adjusts our algorithm to favor our bundled ROCm.
      I've confirmed VRAM reporting still doesn't work properly so we
      can't yet enable concurrency by default.
      1f50356e
  30. 05 Jul, 2024 1 commit
  31. 14 Jun, 2024 1 commit
  32. 04 Jun, 2024 1 commit
  33. 31 May, 2024 1 commit
  34. 30 May, 2024 1 commit
  35. 07 May, 2024 1 commit