1. 14 Feb, 2025 8 commits
    • Jesse Gross's avatar
      ggml-backend: Close on nil should be a no-op · d223f3b6
      Jesse Gross authored
      d223f3b6
    • Jesse Gross's avatar
      ggml-backend: Ensure data is available after async computation · 60830695
      Jesse Gross authored
      We need to sync before retrieving data after async computation.
      It is also important to ensure that the Go buffer is not moved by
      the GC across function calls so we do a synchronous copy.
      60830695
    • Jesse Gross's avatar
      ggml-backend: Let GGML allocate context memory · 01d9a468
      Jesse Gross authored
      Passing in a Go buffer is not safe because the garbage collector could
      free or move the memory while the context is still open. However, if
      we pass in the size and a nil pointer then GGML will allocate it from
      the C side.
      01d9a468
    • Jesse Gross's avatar
      backend: API to support full precision matmul · d773b7d6
      Jesse Gross authored
      Most tensor backends try to optimize performance by using a lower
      precision for matmuls. However, some operations (such as kq) on
      some models are sensitive to this and require full precision.
      d773b7d6
    • Jesse Gross's avatar
      backend: Support graph computation that does not return an output · 4d4463b2
      Jesse Gross authored
      There are two cases where we may not have an output after computing:
       - Prompt processing where the length of the input exceeds the batch
         size
       - Internal memory management operations such as cache defrag and shift
      4d4463b2
    • Jesse Gross's avatar
      backend: Consistently use int (vs. int64) for tensor shapes · 0e38297f
      Jesse Gross authored
      Currently there is a mixture of int and int64 used when dealing with
      tensor dimensions and shapes, which causes unnecessary conversions -
      they all should be the same type.
      
      In general, most interfaces (such as Pytorch) use int64 for
      generality but most implementations (such as CUDA) use int32 for
      performance. There isn't much benefit to us to being more flexible
      than the implementations we are likely to run on.
      
      In addition, as a practical matter, a model with a tensor with a single
      dimension larger than 32 bits is unlikely to run on a 32-bit machine.
      0e38297f
    • Jesse Gross's avatar
      backend: Don't return an error on Close · 7e13f568
      Jesse Gross authored
      It is not common to return errors with close/free operations - most
      people won't check it and even if they did there's probably not much
      that can do. It's better to not give implementations false expectations.
      7e13f568
    • Michael Yang's avatar
      next ollama runner (#7913) · 58245413
      Michael Yang authored
      
      
      feat: add new Ollama engine using ggml through cgo
      
      This change introduces a new way to run pretrained models. It introduces 3 high level interfaces and a bunch of smaller helper interfaces to facilitate this.
      
      - `model.Model` defines the interface for a model architecture. Models such as `llama` and `mllama`, which are provided as examples, can implement the model's forward propagation in the `Forward` method. This method will be called to generate completions. This interface can be found in `model/model.go`
      - `ml.Backend` defines the interface for a backend tensor library, in this case `ggml`. Among other things, a Backend is responsible for loading a pretrained model into hardware (GPU, CPU, etc) and providing an interface for Models to access loaded tensors. This interface can be found in `ml/backend.go`
      - `ml.Tensor` defines the interface for a tensor and tensor operations
      
      This is the first implementation of the new engine. Follow up PRs will implement more features:
      
      - non-greedy sampling (#8410)
      - integration with Ollama and KV caching (#8301)
      - more model support (#9080) with more coming soon
      Co-authored-by: default avatarBruce MacDonald <brucewmacdonald@gmail.com>
      58245413
  2. 13 Feb, 2025 4 commits
  3. 12 Feb, 2025 3 commits
  4. 11 Feb, 2025 2 commits
  5. 10 Feb, 2025 2 commits
  6. 08 Feb, 2025 4 commits
  7. 07 Feb, 2025 6 commits
  8. 06 Feb, 2025 11 commits