transfer_backend

Applicability

Product

Supported (√/x)

Atlas 350 Accelerator Card

x

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

x

Atlas inference product

x

Atlas training product

x

Note: For Atlas A2 training product/Atlas A2 inference product, only the Atlas 800I A2 inference server and A200I A2 Box heterogeneous subrack are supported.

Function Description

Sets the transmission backend used by LLM-DataDist.

Prototype

1
transfer_backend(transfer_backend)

Parameters

Parameter

Data Type

Description

transfer_backend

str

The value is "hixl", indicating that HIXL is used as the transmission engine backend.

Example

1
2
3
from llm_datadist import LLMConfig
llm_config = LLMConfig()
llm_config.transfer_backend = "hixl"

Returns

In normal cases, no value is returned.

If a parameter is incorrect, a TypeError or ValueError may be thrown.

Restrictions

  • When initializing the parameter, you need to specify listen_ip_info. When the hixl transmission backend is used, each transmission end can function as both a client and a server.
  • Before initiating transmission with the peer end, you need to call link_clusters to establish a link.