NVIDIA / JAX-Toolbox

JAX-Toolbox
Apache License 2.0
249 stars 49 forks source link

`xla_gpu_pgle_accuracy_checker is set to WARN, but no profile path specified in xla_gpu_pgle_profile_file_or_directory_path` #1135

Open andportnoy opened 3 weeks ago

andportnoy commented 3 weeks ago

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
andportnoy commented 3 weeks ago

This seems like it should fix it: https://github.com/openxla/xla/pull/18763.