[cpu] Implement all gather/reduce for arm64 cpu (#12527)

2025-12-08 19:03:04 +08:00
parent 4a62a0e3cd
commit 7871593cc8
5 changed files with 419 additions and 290 deletions
--- a/python/pyproject_cpu.toml
+++ b/python/pyproject_cpu.toml
@@ -23,13 +23,13 @@ dependencies = [
  "build",
  "compressed-tensors",
  "datasets",
-  "decord",
+  "decord; platform_machine == 'x86_64'",
  "einops",
  "fastapi",
  "gguf",
  "hf_transfer",
  "huggingface_hub",
-  "intel-openmp",
+  "intel-openmp; platform_machine == 'x86_64'",
  "interegular",
  "llguidance>=0.7.11,<0.8.0",
  "modelscope",