jax.log_compiles

内容

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

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

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