Skip to content

Commit 57c3d9d

Browse files
committed
[TransferEngine]: Introduce generic NVMeoF transport
Signed-off-by: Jinlong Chen <[email protected]>
1 parent b66c445 commit 57c3d9d

22 files changed

+2230
-3
lines changed

dependencies.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,9 @@ SYSTEM_PACKAGES="build-essential \
121121
libcurl4-openssl-dev \
122122
libhiredis-dev \
123123
pkg-config \
124-
patchelf"
124+
patchelf \
125+
libaio-dev \
126+
libnvme-dev"
125127

126128
apt-get install -y $SYSTEM_PACKAGES
127129
check_success "Failed to install system packages"

mooncake-common/common.cmake

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ option(BUILD_EXAMPLES "Build examples" ON)
5858
option(BUILD_UNIT_TESTS "Build unit tests" ON)
5959
option(USE_CUDA "option for enabling gpu features" OFF)
6060
option(USE_NVMEOF "option for using NVMe over Fabric" OFF)
61+
option(USE_NVMEOF_GENERIC "option for using generic NVMe over Fabric transport" OFF)
6162
option(USE_TCP "option for using TCP transport" ON)
6263
option(USE_ASCEND "option for using npu with HCCL" OFF)
6364
option(USE_ASCEND_DIRECT "option for using ascend npu with adxl engine" OFF)
@@ -87,6 +88,11 @@ if (USE_NVMEOF)
8788
message(STATUS "NVMe-oF support is enabled")
8889
endif()
8990

91+
if (USE_NVMEOF_GENERIC)
92+
add_compile_definitions(USE_NVMEOF_GENERIC)
93+
message(STATUS "Generic NVMe-oF support is enabled")
94+
endif()
95+
9096
if (USE_MNNVL)
9197
set(USE_CUDA ON)
9298
add_compile_definitions(USE_MNNVL)

mooncake-transfer-engine/example/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,4 +21,9 @@ endif()
2121
if (USE_ASCEND_DIRECT)
2222
add_executable(transfer_engine_ascend_direct_perf transfer_engine_ascend_direct_perf.cpp)
2323
target_link_libraries(transfer_engine_ascend_direct_perf PUBLIC transfer_engine)
24+
endif()
25+
26+
if (USE_NVMEOF_GENERIC)
27+
add_executable(transfer_engine_nvmeof_generic_bench transfer_engine_nvmeof_generic_bench.cpp)
28+
target_link_libraries(transfer_engine_nvmeof_generic_bench PUBLIC transfer_engine)
2429
endif()

0 commit comments

Comments
 (0)