ollama/model/models/llama4
Daniel Hiltgen 517807cdf2
perf: build graph for next batch async to keep GPU busy (#11863)
* perf: build graph for next batch in parallel to keep GPU busy

This refactors the main run loop of the ollama runner to perform the main GPU
intensive tasks (Compute+Floats) in a go routine so we can prepare the next
batch in parallel to reduce the amount of time the GPU stalls waiting for the
next batch of work.

* tests: tune integration tests for ollama engine

This tunes the integration tests to focus more on models supported
by the new engine.
2025-08-29 14:20:28 -07:00
..
model.go perf: build graph for next batch async to keep GPU busy (#11863) 2025-08-29 14:20:28 -07:00
model_text.go use nn.Linear in place of ml.Tensor (#11049) 2025-06-11 12:10:15 -07:00
model_vision.go ml: Panic rather than return error on tensor allocation failure 2025-05-22 14:38:09 -07:00
process_image.go image processing 2025-04-25 16:59:20 -07:00
process_image_test.go image processing 2025-04-25 16:59:20 -07:00