Skip to content

Commit 47fb17e

Browse files
committed
PLUGINS/UCX: Moved src/utils/ucx/* to src/plugins/ucx.
Signed-off-by: Raul Akhmetshin <[email protected]>
1 parent 04e1216 commit 47fb17e

File tree

17 files changed

+36
-55
lines changed

17 files changed

+36
-55
lines changed

CODEOWNERS

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,8 @@ CODEOWNERS @ai-dynamo/Devops @ai-dynamo/nixl-maintainers
1717
/src/bindings/python @ovidiusm @mkhazraee @roiedanino
1818
/src/bindings/rust @roiedanino @gleon99 @mkhazraee
1919

20-
# UCX Plugins
21-
/src/plugins/ucx* @brminich @yosefe @gleon99
22-
/src/utils/ucx @brminich @yosefe @gleon99
20+
# UCX Plugin
21+
/src/plugins/ucx @brminich @yosefe @gleon99
2322

2423
# Storage Plugins
2524
/src/plugins/posix @w1ldptr @barneuman @etoledano @vvenkates27
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

src/plugins/ucx/meson.build

Lines changed: 27 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,27 +13,46 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515

16-
ucx_utils_dep = declare_dependency(link_with: ucx_utils_lib, include_directories: utils_inc_dirs )
1716
asio_dep = [dependency('asio', required: true)]
1817

1918
compile_flags = []
2019
if cuda_dep.found()
2120
compile_flags = [ '-DHAVE_CUDA' ]
2221
endif
2322

23+
ucx_backend_includes = include_directories('.')
24+
25+
ucx_backend_sources = ['config.cpp',
26+
'gpu_xfer_req_h.cpp',
27+
'rkey.cpp',
28+
'ucx_backend.cpp',
29+
'ucx_plugin.cpp',
30+
'ucx_utils.cpp']
31+
32+
ucx_backend_dependencies = [asio_dep,
33+
cuda_dep,
34+
nixl_common_dep,
35+
nixl_infra,
36+
serdes_interface,
37+
thread_dep,
38+
ucx_dep]
39+
40+
ucx_backend_include_directories = [nixl_inc_dirs,
41+
utils_inc_dirs]
42+
2443
if 'UCX' in static_plugins
2544
ucx_backend_lib = static_library('UCX',
26-
'ucx_backend.cpp', 'ucx_backend.h', 'ucx_plugin.cpp',
27-
dependencies: [nixl_infra, ucx_utils_dep, serdes_interface, cuda_dep, ucx_dep, thread_dep, nixl_common_dep, asio_dep],
28-
include_directories: nixl_inc_dirs,
45+
ucx_backend_sources,
46+
dependencies: ucx_backend_dependencies,
47+
include_directories: ucx_backend_include_directories,
2948
install: false,
3049
cpp_args : compile_flags,
3150
name_prefix: 'libplugin_') # Custom prefix for plugin libraries
3251
else
3352
ucx_backend_lib = shared_library('UCX',
34-
'ucx_backend.cpp', 'ucx_backend.h', 'ucx_plugin.cpp',
35-
dependencies: [nixl_infra, ucx_utils_dep, serdes_interface, cuda_dep, ucx_dep, thread_dep, nixl_common_dep, asio_dep],
36-
include_directories: nixl_inc_dirs,
53+
ucx_backend_sources,
54+
dependencies: ucx_backend_dependencies,
55+
include_directories: ucx_backend_include_directories,
3756
install: true,
3857
cpp_args : compile_flags + ['-fPIC'],
3958
name_prefix: 'libplugin_', # Custom prefix for plugin libraries
@@ -48,4 +67,4 @@ else
4867
endif
4968
endif
5069

51-
ucx_backend_interface = declare_dependency(link_with: ucx_backend_lib)
70+
ucx_backend_interface = declare_dependency(link_with: ucx_backend_lib, include_directories: ucx_backend_includes)
File renamed without changes.
File renamed without changes.

src/plugins/ucx/ucx_backend.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
#include "common/nixl_log.h"
2020
#include "serdes/serdes.h"
2121
#include "common/nixl_log.h"
22-
#include "ucx/gpu_xfer_req_h.h"
22+
#include "gpu_xfer_req_h.h"
2323

2424
#include <optional>
2525
#include <limits>

src/plugins/ucx/ucx_backend.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@
3535

3636
// Local includes
3737
#include "common/nixl_time.h"
38-
#include "ucx/rkey.h"
39-
#include "ucx/ucx_utils.h"
38+
#include "rkey.h"
39+
#include "ucx_utils.h"
4040

4141
enum ucx_cb_op_t { NOTIF_STR };
4242

0 commit comments

Comments
 (0)