xmax1 comments

Repositories
Issues
Comments

Results 2 comments of


                                            xmax1

Allowing pmap on CPUs by making CPU cores visibles directly on Jax.

What is the status on this and how should using the XLA_FLAGS="--xla_force_host_platform_device_count=200" affect the behaviour? Looking to test pmap on multiple cpus.

No kernel image is available for execution on the device

Getting the same error and the XLA_FLAGS=--xla_gpu_force_compilation_parallelism=1 doesn't work for me and introduces a new error. jax version ` packages in environment at /home/energy/amawi/miniconda3/envs/pansatz: Name Version Build Channel jax 0.2.24...