jax.log_compiles

jax.log_compiles = <jax._src.config.State 对象>

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

每当jit或pmap编译XLA计算时，记录一条消息。日志通过logging模块进行。当此选项启用时，日志级别为WARNING；否则为DEBUG。

参数:

new_val (任意类型)