xla icon indicating copy to clipboard operation
xla copied to clipboard

[GPU] Expose caching DebugOptions to python

Open trevor-m opened this issue 1 year ago • 2 comments

Adds python bindings for xla_gpu_kernel_cache_file, xla_gpu_enable_llvm_module_compilation_parallelism and xla_gpu_per_fusion_autotune_cache_dir.

We would like to add some convenience features to JAX which will enable all caches with one flag/option (will open PR for that soon). This change is necessary for that.

trevor-m avatar Aug 01 '24 19:08 trevor-m

@skye @hawkinsp for PjRT API changes

cheshire avatar Aug 02 '24 11:08 cheshire

Is it possible to test those?

Thanks for reviewing, added test to xla_client_test.

trevor-m avatar Aug 05 '24 21:08 trevor-m

What can we do to get this PR merged? @cheshire @skye

nouiz avatar Aug 19 '24 18:08 nouiz