create_cpu_cache

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

Creates a CPU cache.

Prototype

create_cpu_cache(cache_desc: CacheDesc, addrs: List[int])

Parameters

Parameter	Data Type	Value Description
cache_desc	CacheDesc	Description of the cache.
addrs	List[int]	Address of the CPU cache.

Example

from llm_datadist import Cache
cpu_cache = Cache.create_cpu_cache(cpu_cache_desc, cpu_addrs) # cpu_addrs is obtained from the created CPU tensors.

Returns

In normal cases, a CPU cache of the cache type is returned.

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

If the input parameter is None, the AttributeError exception is reported.

Constraints

None

Parent topic: Cache