Add nccl as dep for aarch64.

trivialfis · trivialfis · commit ff1117d29359 · 2025-10-16T05:25:17.000+08:00
diff --git a/ops/script/pypi_variants.py b/ops/script/pypi_variants.py
@@ -11,7 +11,7 @@
 STUB_IN_PATH = os.path.join(PY_PACKAGE, "pyproject.toml.stub.in")
 OUT_PATH = os.path.join(PY_PACKAGE, "pyproject.toml")
 
-NCCL_WHL = """    \"nvidia-nccl-{0} ; platform_system == 'Linux' and platform_machine != 'aarch64'\","""
+NCCL_WHL = """    \"nvidia-nccl-{0} ; platform_system == 'Linux'\","""
 
 NAME = "{{ name }}"
 NCCL = "{{ nccl }}"
diff --git a/python-package/pyproject.toml b/python-package/pyproject.toml
@@ -33,7 +33,7 @@ classifiers = [
 dependencies = [
     "numpy",
     "scipy",
-    "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'",
+    "nvidia-nccl-cu12 ; platform_system == 'Linux'",
 ]
 
 [project.urls]

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@ classifiers = [`
`33`	`33`	`dependencies = [`
`34`	`34`	`"numpy",`
`35`	`35`	`"scipy",`
`36`		`- "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'",`
	`36`	`+ "nvidia-nccl-cu12 ; platform_system == 'Linux'",`
`37`	`37`	`]`
`38`	`38`
`39`	`39`	`[project.urls]`