jax.experimental.compilation_cache 模組#

JAX 磁碟編譯快取。

API#

jax.experimental.compilation_cache.compilation_cache.is_initialized()[原始碼]#

已棄用。

傳回是否已啟用快取。初始化可以延遲，因此不檢查初始化狀態。名稱保留是為了向後相容性。

傳回類型:

bool

jax.experimental.compilation_cache.compilation_cache.initialize_cache(path)[原始碼]#

此 API 已棄用；請改用 set_cache_dir。

設定路徑。若要生效，應在任何呼叫 get_executable_and_time() 和 put_executable_and_time() 之前呼叫。

傳回類型:

None

jax.experimental.compilation_cache.compilation_cache.set_cache_dir(path)[原始碼]#

設定持久性編譯快取目錄。

呼叫此函式後，jit 編譯的函式會儲存到 path，因此如果重新啟動程序或再次執行，則不需要重新編譯。這也會告知 Jax 在編譯之前應於何處尋找已編譯的函式。

傳回類型:

None

jax.experimental.compilation_cache.compilation_cache.reset_cache()[原始碼]#

回到初始、未初始化的狀態。

傳回類型:

None