BlocksCacheKey

Applicable Products

Product

Supported (√/x)

Ascend 950PR/Ascend 950DT

√

Atlas A3 training products/Atlas A3 inference products

√

Atlas A2 training products/Atlas A2 inference products

√

Atlas 200I/500 A2 inference products

x

Atlas inference products

x

Atlas training products

x

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

Function Description

Constructs a BlocksCacheKey instance, which is usually used as the parameter type in allocate_blocks_cache, pull_blocks, and push_blocks of CacheManager.

Prototype

1
__init__(*args, **kwargs)

Parameters

Parameter

Data Type

Description

cluster_id

int

(Mandatory) ID of the remote cluster where the cache is located.

model_id

int

ID of the model associated with the cache. The default value is 0.

Example

1
2
from llm_datadist import BlocksCacheKey
blocks_cache_key = BlocksCacheKey(0, 0)

Returns

In normal cases, a BlocksCacheKey instance is returned.

If the input data type is incorrect, the TypeError or ValueError exception is reported.

Constraints

None