From 0163a626213e037a5a8c93d0184c4a5072aa1fe8 Mon Sep 17 00:00:00 2001
From: Nicolas Richart <networms@gmail.ch>
Date: Thu, 20 Jun 2024 23:17:43 +0200
Subject: [PATCH] UCX more variants

---
 stacks/pinot-noir/packages_jed.yaml   | 6 ++++++
 stacks/pinot-noir/packages_stack.yaml | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/stacks/pinot-noir/packages_jed.yaml b/stacks/pinot-noir/packages_jed.yaml
index 7169252..155ace9 100644
--- a/stacks/pinot-noir/packages_jed.yaml
+++ b/stacks/pinot-noir/packages_jed.yaml
@@ -12,3 +12,9 @@ packages:
         when: '^cuda'
 
       - '+cuda'
+
+  ucx:
+    require: [+xpmem, +knem, +cma,
+     +rdmacm, +rc, +ud, +dc, +verbs, +ib_hw_tm,
+     +vfs,
+     +gdrcopy, +cuda]
diff --git a/stacks/pinot-noir/packages_stack.yaml b/stacks/pinot-noir/packages_stack.yaml
index ff5ba6b..0ba8329 100644
--- a/stacks/pinot-noir/packages_stack.yaml
+++ b/stacks/pinot-noir/packages_stack.yaml
@@ -62,7 +62,7 @@ packages:
 
   ucx:
     require:
-    - spec: +rdmacm +rc +dc +ud +cma +verbs
+    - spec: +rdmacm +rc +dc +ud +cma +verbs +vfs
     - spec: +gdrcopy
       when: +cuda
 
-- 
GitLab