jax.log_compiles

jax.log_compiles = <jax._src.config.State object>

用於 jax_log_compiles 設定選項的上下文管理器。

每次 jit 或 pmap 編譯 XLA 計算時，記錄訊息。記錄使用 logging 執行。當設定此選項時，日誌層級為 WARNING；否則層級為 DEBUG。

參數:

new_val (Any)