From b4d12025741c0e1eaa05ca75521223ad8ce388a6 Mon Sep 17 00:00:00 2001
From: ferres <justferres@yandex.ru>
Date: Fri, 11 Jul 2025 02:07:04 +0300
Subject: [PATCH] python3Packages.xgrammar: add aarch64-darwin support

Update pkgs/development/python-modules/xgrammar/default.nix

Co-authored-by: Sarah Clark <seclark@nextquestion.net>
---
 .../python-modules/xgrammar/default.nix       | 21 ++++++++-----------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/pkgs/development/python-modules/xgrammar/default.nix b/pkgs/development/python-modules/xgrammar/default.nix
index 6a1a1549293fd..8ada40b95593d 100644
--- a/pkgs/development/python-modules/xgrammar/default.nix
+++ b/pkgs/development/python-modules/xgrammar/default.nix
@@ -11,14 +11,13 @@
   scikit-build-core,
 
   # dependencies
-  mlx-lm,
   pydantic,
   sentencepiece,
   tiktoken,
   torch,
   transformers,
   triton,
-
+  mlx-lm,
   # tests
   pytestCheckHook,
   writableTmpDirAsHomeHook,
@@ -56,22 +55,20 @@ buildPythonPackage rec {
     torch
     transformers
   ]
-  ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isx86_64) [
-    triton
-  ]
-  ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isAarch64) [
-    mlx-lm
-  ];
+  ++ lib.optional (lib.meta.availableOn stdenv.hostPlatform triton) triton
+  ++ lib.optional (lib.meta.availableOn stdenv.hostPlatform mlx-lm) mlx-lm;
 
   nativeCheckInputs = [
     pytestCheckHook
     writableTmpDirAsHomeHook
   ];
 
-  NIX_CFLAGS_COMPILE = lib.optionalString stdenv.hostPlatform.isLinux (toString [
-    # xgrammar hardcodes -flto=auto while using static linking, which can cause linker errors without this additional flag.
-    "-ffat-lto-objects"
-  ]);
+  NIX_CFLAGS_COMPILE = toString (
+    lib.optionals stdenv.cc.isGNU [
+      # xgrammar hardcodes -flto=auto while using static linking, which can cause linker errors without this additional flag.
+      "-ffat-lto-objects"
+    ]
+  );
 
   disabledTests = [
     # You are trying to access a gated repo.