jax.log_compiles

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

jax_log_compiles 配置选项的上下文管理器。

每次 jit 或 pmap 编译 XLA 计算时记录一条消息。日志记录使用 logging 执行。设置此选项时，日志级别为 WARNING；否则级别为 DEBUG。

参数:

new_val (Any)