Running anything in JAX using today's container prints warnings:
import jax.numpy as jnp
2 * jnp.ones(1)
Output:
2024-10-31 21:29:38.149538: W external/xla/xla/service/gpu/gpu_hlo_schedule.cc:473] xla_gpu_pgle_accuracy_checker is set to WARN, but no profile path specified in xla_gpu_pgle_profile_file_or_directory_path
2024-10-31 21:29:38.836258: W external/xla/xla/service/gpu/gpu_hlo_schedule.cc:473] xla_gpu_pgle_accuracy_checker is set to WARN, but no profile path specified in xla_gpu_pgle_profile_file_or_directory_path
2024-10-31 21:29:38.847071: W external/xla/xla/service/gpu/gpu_hlo_schedule.cc:473] xla_gpu_pgle_accuracy_checker is set to WARN, but no profile path specified in xla_gpu_pgle_profile_file_or_directory_path
Running anything in JAX using today's container prints warnings:
Output: