diff --git a/flake.nix b/flake.nix
index 6d108d7c5..a353b8c3b 100644
--- a/flake.nix
+++ b/flake.nix
@@ -21,6 +21,9 @@
           buildInputs = with pkgs; lib.optionals stdenv.isDarwin [
             darwin.apple_sdk.frameworks.Accelerate
           ];
+          makeFlags = with pkgs; lib.optionals (system == "aarch64-darwin") [
+            "CFLAGS=-D__ARM_FEATURE_DOTPROD=1"
+          ];
           buildPhase = ''
             make main quantize quantize-stats perplexity embedding vdot libllama.so
           '';