diff --git a/stacks/pinot-noir/packages_jed.yaml b/stacks/pinot-noir/packages_jed.yaml index 7169252d5dcf127f2a975eedfe880d026443f4af..155ace969c128f393fb266657a0e72c4da6927b7 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 ff5ba6b4bf0ed7aadedfd3c918ae7e696ccf9926..0ba83299b023adf4cbb4eacfa1dc1379fe92a23a 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