SciML / PSOGPU.jl

GPU accelerated Particle Swarm Optimization
MIT License
13 stars 1 forks source link

Try PINN example #27

Open Vaibhavdixit02 opened 9 months ago

Vaibhavdixit02 commented 9 months ago

Currently on CPU it was terrible compared to BFGS but I think on GPU it should show better performance. Couldn't run it there yet because supercloud GPU access is weird have to figure it out.

utkarsh530 commented 9 months ago

Yes, getting this working with GPU would require some effort to make everything compatible. (ComponentArrays as Array types won't work within structs 😓 )

utkarsh530 commented 9 months ago

Can you show your benchmarks on CPU as well?

Vaibhavdixit02 commented 9 months ago

It was too slow on my computer I had to interrupt it