dgl.graphbolt.cpu_cached_feature

dgl.graphbolt.cpu_cached_feature(fallback_features: Feature | Dict[FeatureKey, Feature], max_cache_size_in_bytes: int, policy: str | None = None, pin_memory: bool = False) → CPUCachedFeature | Dict[FeatureKey, CPUCachedFeature]

包装后备特征的 CPU 缓存特征。

参数：

• fallback_features (Union[Feature, Dict[FeatureKey, Feature]]) – 后备特征。

• max_cache_size_in_bytes (int) – 缓存容量，以字节为单位。此大小应比每个读取请求的大小大几倍。否则，缓存策略可能会因所有缓存条目都被读/写锁定而挂起，导致死锁。

• policy (str, optional) – 缓存淘汰策略算法名称。可用策略包括 [“s3-fifo”, “sieve”, “lru”, “clock”]。默认为 “sieve”。

• pin_memory (bool, optional) – 缓存存储是否应分配在系统固定内存上。默认为 False。

返回值：

包装了 CPUCachedFeature 的新特征。

返回类型：

Union[CPUCachedFeature, Dict[FeatureKey, CPUCachedFeature]]