search

mattloper / chumpy

MIT License
195 stars 118 forks source link

Optimizing for GPU #44

Open ulucsahin opened 3 years ago

ulucsahin commented 3 years ago

Hello, first of all thank you for the useful library.

I see that computations are running on the CPU. In my work this causes a speed bottleneck and I would like to run these computations, or at least some parts on GPU. I was examining the code and my code uses minimize_dogleg function from ch.optimization. Lines with state.updateJ is where most of the is being spent.

I was trying to convert some operations to GPU using Numba library but I don't have much success so far. Does anyone tried this before, or if you can show some directions that would be appreciated.

ritaank commented 1 year ago

were you able to achieve any success here ?

martin-wmx commented 1 year ago

Hi, I am facing the same problem, do you have a solution now?