jax.experimental.compilation_cache
模块#
JAX 磁盘编译缓存。
API#
- jax.experimental.compilation_cache.compilation_cache.is_initialized()[source]#
已弃用。
返回缓存是否已启用。初始化可以延迟,因此不会检查初始化状态。该名称保留是为了向后兼容。
- 返回类型:
- jax.experimental.compilation_cache.compilation_cache.initialize_cache(path)[source]#
此 API 已弃用;请改用 set_cache_dir。
设置路径。要生效,应在调用 get_executable_and_time() 和 put_executable_and_time() 之前调用。
- 返回类型:
None