From e3809c7e7f56281b319c2c4d2e60077016c4b051 Mon Sep 17 00:00:00 2001
From: Yang Liwei <yang0345@e.ntu.edu.sg>
Date: Wed, 26 Nov 2025 18:26:55 +0800
Subject: [PATCH] Add nvidia-nccl-cu12 library to poetry.

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 65524c0..01a8630 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,6 +25,7 @@ qibo = "^0.2.16"
 quimb = { version = "^1.10.0", extras = ["tensor"] }
 cupy-cuda12x = { version = "^13.6.0", optional = true }
 cuda-toolkit = {extras = ["all"], version = "^12.9.1"}
+nvidia-nccl-cu12 = { version = "^2.16.5", optional = true }
 cuquantum-python-cu12 = { version = "^25.9.1", optional = true }
 qmatchatea = { version = "^1.4.3", optional = true }
 qiskit = { version = "^1.4.0", optional = true }
@@ -32,7 +33,7 @@ qtealeaves = { version = "^1.5.20", optional = true }
 
 
 [tool.poetry.extras]
-cuda = ["cupy-cuda12x", "cuda-toolkit", "cuquantum-python-cu12", "mpi4py"]
+cuda = ["cupy-cuda12x", "cuda-toolkit", "nvidia-nccl-cu12", "cuquantum-python-cu12", "mpi4py"]
 qmatchatea = ["qmatchatea"]
 
 [tool.poetry.group.docs]