pylibwholegraph.torch.embedding.create_wholememory_cache_policy

pylibwholegraph.torch.embedding.create_wholememory_cache_policy(cache_comm: WholeMemoryCommunicator, *, memory_type: str = 'chunked', memory_location: str = 'cuda', access_type: str = 'readonly', ratio: float = 0.5)

创建WholeMemoryCachePolicy 注意：在大多数情况下，create_builtin_cache_policy()

可以支持。此函数是一个更灵活的接口

Parameters:

• cache_comm – 缓存的WholeMemory通信器

• memory_type – WholeMemory 缓存类型

• memory_location – 缓存的WholeMemory位置

• access_type – 所需的访问类型

• ratio – 缓存比率

Returns:

WholeMemoryCachePolicy