Merge pull request #687 from boegel/better_software_overview

pick up on GPU info for software overview page

Author: ocaisa

.github/workflows/test_overview_available_software.yml

@@ -25,4 +25,5 @@ jobs:
           . venv/bin/activate
           pip install -r mkdocs-ldjson-plugin/requirements.txt
 
-          python .github/workflows/scripts/test_overview_available_software.py
+          # disabled, this needs to be re-worked after the changes implemented in https://github.com/EESSI/docs/pull/677
+          # python .github/workflows/scripts/test_overview_available_software.py

docs/available_software/index.md

@@ -7,12 +7,6 @@ Overview of software available in [EESSI's production repository `software.eessi
 
 <em>{{ data.n_software }} unique software projects (+ {{ data.n_extensions }} unique extensions)</em>
 
-!!! warning "Details on GPU installations not included yet"
-
-    The overview currently doesn't expose information about GPU installations of software yet.
-
-    This will be resolved soon.
-
 <!-- see also docs/available_software/javascripts/software-filter.js -->
 <input type="search" id="software-search" class="md-input"
        placeholder="Filter software..."

docs/available_software/macros.py

@@ -15,6 +15,12 @@
     'aarch64': ['Arm'],
     'riscv64': ['RISC-V'],
 }
+GPU_ARCHS = {
+    'amd': 'AMD',
+    'accel/amd': 'AMD',
+    'nvidia': 'NVIDIA',
+    'accel/nvidia': 'NVIDIA',
+}
 
 
 def define_env(env):
@@ -53,7 +59,18 @@ def load_json_eessi_software():
             licenses = ', '.join(y for x in versions for y in x['license'])
 
             # determine set of supported CPU families (first part of CPU target names, like x86_64 or aarch64)
-            cpu_families = set(c for v in versions for x in v['cpu_arch'] for c in CPU_ARCHS[x.split('/')[0]])
+            cpu_families = set()
+            for version in versions:
+                for cpu_arch in version['cpu_arch']:
+                    cpu_family = cpu_arch.split('/')[0]
+                    cpu_families.union(CPU_ARCHS.get(cpu_family, cpu_family))
+
+            # determine set of supported GPU families
+            gpu_families = set()
+            for version in versions:
+                for gpu_arch in [y for x in version['gpu_arch'].values() for y in x]:
+                    gpu_family = '/'.join(gpu_arch.split('/')[:-1])
+                    gpu_families.add(GPU_ARCHS.get(gpu_family, gpu_family))
 
             # EESSI versions in which this software is available
             eessi_versions = set()
@@ -65,7 +82,7 @@ def load_json_eessi_software():
                 'n_versions': len(versions),
                 'licenses': licenses,
                 'cpu_families': ', '.join(x for x in sorted(cpu_families)),
-                'gpu_families': '',
+                'gpu_families': ', '.join(x for x in sorted(gpu_families)),
                 'is_extension': False,
             }