Mehdi Ataei comments

Results 15 comments of


                                            Mehdi Ataei

When I run a local model, GPU is used, but the CPU is 100%

Same issue. GPU is not used at all (the memory is allocated though)

Using CUDA, but GPU shows near 0% usage

Same issue when using 2x RTX 6000 Ada gen.

Make Pallas/GPU easier to install

Yes please!

JIT+Shardmap+donate_argnums doesn't work, while pmap+donate_argnums works.

Hey @mattjj any chance to take a look at this? Thanks.

Reopen previous ZED state, while opening a new file

@fred-sheehan Not actually. It keeps opening the project even after removing it!

The demo occurs in paper

Hello @whpy Currently, we are undergoing a significant refactoring process, which can be tracked through our branches. We expect to complete this refactoring soon and it will include all the...

The demo occurs in paper

Yes you can. Please use the latest 4.30 version.

Apple Silicon: error: failed to legalize operation 'mhlo.custom_call'

Anyway to ignore this in the code without removing the shardings?

Apple Silicon: error: failed to legalize operation 'mhlo.custom_call'

I think c is the best option, and I have noticed small performance gains if I explicitly remove shardings when using 1 device. P.S. I am not sure if it...

IndicesBoundaryMasker in JAX causes halting problem when used in multi-GPU

Just run any example (e.g., flow over sphere) multiple times with having both GPUs visible. It will halt 90% of the time. Please use the old implementation to fix this....