Skip to content

Commit 605c087

Browse files
committed
[None][feat] Add NIXL-LIBFABRIC
Allows users to use LIBFABRIC as NIXL backend for KV cache transmission. Usage: TRTLLM_NIXL_KVCACHE_BACKEND=LIBFABRIC # (or UCX) Currently supports only UCX and LIBFABRIC as NIXL KV cahce backends. Unsupported backend types will fallback to Default (UCX). Signed-off-by: Yoray Zack <[email protected]>
1 parent 1d6fbbf commit 605c087

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cpp/tensorrt_llm/executor/cache_transmission/nixl_utils/transferAgent.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -348,11 +348,11 @@ NixlTransferAgent::NixlTransferAgent(BaseAgentConfig const& config)
348348

349349
std::string nixlBackend = common::getEnvNixlBackend();
350350
// List of supported backends - extend this list as new backends are added
351-
static const std::set<std::string> kSUPPORTED_BACKENDS = {"UCX"};
351+
static const std::set<std::string> kSUPPORTED_BACKENDS = {"UCX", "LIBFABRIC"};
352352

353353
if (kSUPPORTED_BACKENDS.find(nixlBackend) == kSUPPORTED_BACKENDS.end())
354354
{
355-
TLLM_LOG_ERROR("Unsupported NIXL backend: %s, fallback to UCX", nixlBackend.c_str());
355+
TLLM_LOG_WARNING("Unsupported NIXL backend: %s, fallback to UCX", nixlBackend.c_str());
356356
nixlBackend = "UCX";
357357
}
358358

0 commit comments

Comments
 (0)