Settings

Theme

Apple Silicon and macOS VMs: Faster LLM Inference with llama.cpp

github.com

307 points by frabonacci · 53 comments

Reader

10 threads
simonw

It looks to me like this won't speed up llama.cpp for everyone, just for users running it in this particular kind of Virtualization.framework VM.

The fix here works around a problem where the VM was causing llama.cpp to select the wrong kernels.

  • frabonacciOP

    > this won't speed up llama.cpp for everyone, just for users running it in this particular kind of Virtualization.framework VM.

    correct. these figures apply to llama.cpp inside the macOS guest configuration we tested. Lume is the VM frontend we used, while Apple's Virtualization.framework provides the virtual GPU. bare-metal llama.cpp is unaffected.

    > The fix here works around a problem where the VM was causing llama.cpp to select the wrong kernels.

    mostly, with one nuance: llama.cpp is selecting the correct kernels for the capability answers it receives. the stock guest reports an older Apple GPU family and a 32 KB threadgroup memory limit, so llama.cpp chooses slower kernels. Our process-scoped layer reports the tested Apple 9 and 64 KB values while allowing llama.cpp to select newer paths that the paravirtual GPU successfully execute

    the layer itself though works at the Metal API boundary, independently of llama.cpp. other Metal compute and graphics apps now may select newer paths from the same capability answers, although this is still preliminary and each app needs separate testing. for example, MLX-LM stayed flat in our tests

    historically related limitations have been coming up across Apple Silicon VM frontends for a while e.g. Tart tracked MPS/GPU support back in 2023: - https://github.com/openai/tart/issues/501 - https://github.com/openai/tart/issues/1032

    UTM also has related cases where apps detect the Apple paravirtual Metal device but falls back to software rendering: https://github.com/utmapp/UTM/issues/7671

    • sitkack

      why do use ai to write your posts ?

      • Aldo_MX

        Why do you expect an AI engineer to manually write prose?

        • b112

          Don't post generated text or AI-edited text. HN is for conversation between humans.

          Because it is not allowed here, that's why. See the guidelines.

        • throwaway173axp

          Why do you think an AI engineer would go through the trouble lower casing everything except for proper nouns and abbreviations?

  • octocop

    a win is still a win

engzaanin

That makes sense. The title initially sounded like a general llama.cpp speedup on Apple Silicon, but if the improvement comes from fixing kernel selection inside Virtualization.framework VMs, that distinction is pretty important.

thehamkercat

> 11.08× faster and generated tokens 16.36× faster than the same workload in the same stock VM.

So this was the comparison, for me the title was a bit confusing

  • frabonacciOP

    yeah fair point. it's always tricky to get the whole idea across within HN's title limit. tldr: we ran the same workload in the same Lume macOS VM on the same Apple Silicon host, first with stock Metal capability reporting and then with our process-scoped dynamic library. The 11.08x figure is prompt processing, while 16.36x is token generation. the mechanism technically extends to graphics workloads too but these figures are specifically from llama.cpp

aeriose

What I don't get, which this article doesn't talk about, why would Apple’s Virtualization.framework expose a lesser Metal profile instead of reporting all capabilities supported by the host GPU?

  • hugmynutus

    Because nobody knows.

    Apple doesn't let you "pass" the GPU through to a VM like most other ARM/x86_64 processors (forwarding interrupts and PCIe memory regions). There are symbols defined to do this within the kernel (if you dump the binary) but they aren't used in retail macos.

    Instead you end up creating a paravirtual device that emulates the GPU acting like a 'normal PCI device' which you give to clients. This is usually reserved (by other hardware vendors) for when you're doing multi-tenat time sharing of higher end GPUs (like Nvidia enterprise cards can do).

    These paravirtualized GPUs then just have 'less features' and Apple (being Apple) states no reason why.

    • frabonacciOP

      my guess is apple chose a conservative profile for compatibility across different chips and guest releases

  • chorizo

    All M-series chips support Metal 4. Wonder if we can fix this with a simple override somewhere.

  • b112

    QEMU/kvm does this as a default, because keeping a more generic CPU / etc makes moving VMs between machines with different hardware possible. If you try to move a VM it won't work, of course, if the new machine doesn't support what the old did.

    Not sure of this is why Apple does it. With KVM, you tend to pick a baseline that all your machines support.

  • bestham

    Because it cannot be safely virtualised?

azinman2

I don’t understand what Apple 1-9 are. At first I thought it was M series chips but there is no M9 (yet)

gigatexal

All this work to get the Mac to be a platform useful for AI is being done despite Apple's efforts. They're famously pissed at Nvidia since the Nvidia + Intel Macs due to heat and other issues. But then the OS is a bit closed off and they move slow and are more focused on milking iOS and the App Store and services for money BUT the PA-Semi purchase and Apple Silicon and everything following it has made the hardware just so amazing and useful that despite all that people build for it.

I love the platform. I'm happy to see people building on it.

AND! if we ever get an M7 chip with the rumored 1.5TB of available ram all this work will not have been in vain. You think the ai acceleration is nice in the M5 wait till M7 and M8.

  • frabonacciOP

    apple silicon is what made us start Lume in the first place last year. the hardware is so good (M1 is now 6 years old!) that people keep pushing through the gaps in the platform. just yesterday we ran a fully offline computer-use agent with Cua Driver and Muse Glimmer, all locally on Apple Silicon, an now the same kind of agent can run isolated inside a macOS VM and use Apple’s GPU path too

shay_ker

I recall there was another YC startup that was working on Mac-specific ML optimizations for local inference (and perhaps fine-tuning).

I wonder if their work is related?

wyzer

I see M1 Ultra host mentioned, are there any M1 pro or M3 pro results? Has anyne tried?

  • frabonacciOP

    We've also seen similar improvements on a M5 max. no M1 pro or M3 pro results yet though - would love to see someone try those

w10-1

Related: does anyone have a basis for guessing whether the Neural Accelerators found in M5 Pro+ (accessed by Metal 4) will make their way into the M6 base processors?

woadwarrior01

The Claudish in the blogpost makes it really hard to ready. Also, TinyLlama 1.1B lol.

Keyboard Shortcuts

j
Next item
k
Previous item
o / Enter
Open selected item
?
Show this help
Esc
Close modal / clear selection