Removed unused env variables and format

axonn-ai · May 8, 2024 · 88176e9 · 88176e9
1 parent f3a5dfc
commit 88176e9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/.github/workflows/nvidia-rtx-3090-tests.yaml b/.github/workflows/nvidia-rtx-3090-tests.yaml
@@ -17,10 +17,6 @@ jobs:
         ginter: [ 1, 2 ]
         memopt: [ '0', '1' ]
 
-    env:
-      SLURM_NTASKS: 0
-      SLURM_PROCID: 0
-
     steps:
     - uses: actions/checkout@v3
     - name: Install AxoNN
@@ -44,10 +40,6 @@ jobs:
   intra-layer:
     runs-on: [ nvidia ]
 
-    env:
-      SLURM_NTASKS: 0
-      SLURM_PROCID: 0
-
     steps:
     - uses: actions/checkout@v3
     - name: Install AxoNN

diff --git a/axonn/axonn.py b/axonn/axonn.py
@@ -166,7 +166,7 @@ def init(
     if device == "cpu":
         assert (
             G_intra_d == 1
-        ), "G_intra_d > 1: Intra_d uses reduce-scatters which gloo for cpus doesn't support"
+        ), "G_intra_d > 1: Intra_d uses reduce-scatters which gloo(cpu) doesn't support"
 
     if mixed_precision:
         computation_dtype = torch.float16