From 02e44e8f7e4a18f7715d2b252e9c12649ebbdffe Mon Sep 17 00:00:00 2001
From: Masayuki Takagi <kamonama@gmail.com>
Date: Thu, 15 Jul 2021 15:16:47 +0000
Subject: [PATCH] NCCL v2.10.3

---
 docker.py | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/docker.py b/docker.py
index d4a8478..604a5e2 100644
--- a/docker.py
+++ b/docker.py
@@ -72,6 +72,7 @@
     'nccl2.7-cuda101', 'nccl2.7-cuda102', 'nccl2.7-cuda110', 'nccl2.7-cuda111',
     'nccl2.8-cuda112',
     'nccl2.9-cuda102', 'nccl2.9-cuda110', 'nccl2.9-cuda113',
+    'nccl2.10-cuda102', 'nccl2.10-cuda110',
 ]
 cutensor_choices = [
     'none',
@@ -116,8 +117,9 @@
     'cuda101': ['nccl2.4-cuda101', 'nccl2.5-cuda101', 'nccl2.6-cuda101',
                 'nccl2.7-cuda101'],
     'cuda102': ['nccl2.5-cuda102', 'nccl2.6-cuda102', 'nccl2.7-cuda102',
-                'nccl2.9-cuda102'],
-    'cuda110': ['nccl2.7-cuda110', 'nccl2.9-cuda110'],
+                'nccl2.9-cuda102', 'nccl2.10-cuda102'],
+    'cuda110': ['nccl2.7-cuda110', 'nccl2.9-cuda110', 'nccl2.10-cuda110',
+                'nccl2.10-cuda110'],
     'cuda111': ['nccl2.7-cuda111'],
     'cuda112': ['nccl2.8-cuda112'],
     'cuda113': ['nccl2.9-cuda113'],
@@ -964,6 +966,24 @@ def get_scipy_choices():
     lib_dir='/usr/lib/x86_64-linux-gnu',
 )
 
+codes['nccl2.9-cuda102'] = nccl_base.format(
+    nccl_sub_dir='cuda',
+    nccl_os_ver='1804',
+    libnccl2='libnccl2_2.10.3-1+cuda10.2_amd64',
+    libnccl_dev='libnccl-dev_2.10.3-1+cuda10.2_amd64',
+    include_dir='/usr/include',
+    lib_dir='/usr/lib/x86_64-linux-gnu',
+)
+
+codes['nccl2.9-cuda110'] = nccl_base.format(
+    nccl_sub_dir='cuda',
+    nccl_os_ver='1804',
+    libnccl2='libnccl2_2.10.3-1+cuda11.0_amd64',
+    libnccl_dev='libnccl-dev_2.10.3-1+cuda11.0_amd64',
+    include_dir='/usr/include',
+    lib_dir='/usr/lib/x86_64-linux-gnu',
+)
+
 # cuTENSOR
 # The shell script needs to be saved in an env var due to Dockerfile limitations
 codes['cutensor1.2.0-cuda101'] = 'RUN eval $CUTENSOR_INSTALL && install_cutensor 1.2.0.2 1.2.0 10.1;'