chunkdot icon indicating copy to clipboard operation
chunkdot copied to clipboard

Adding GPU Support

Open sickerin opened this issue 1 year ago • 4 comments

Hi, great project! Are you working on making or have made GPU support possible? I saw a closed issue from a while ago. I looked through the code, but I don't think I see it implemented yet. I might be interested in giving it a go! Lmk if there's any potential issues that you found, etc if you have already tried adding GPU support previously

sickerin avatar Jan 20 '24 12:01 sickerin

Hi @sickerin, I have not tried to add GPU support. I would really appreciate a contribution for that. I can try to help if needed. Are you familiar with Numbat GPU acceleration features? I think that would be the place to start.

rragundez avatar Jan 21 '24 01:01 rragundez

Hi @sickerin, what do you think? Do you want to give it a try?

rragundez avatar Jan 31 '24 11:01 rragundez

Hi yep, I'll give it a try sometime this weekend or the next. I'm actually not so familiar with Numba, but I think it'd be useful to know. I did try some simple @cuda.jit stuff referencing this https://numba.readthedocs.io/en/stable/cuda/examples.html#matrix-multiplication, but it didn't work, so I'll have a closer look.

sickerin avatar Jan 31 '24 12:01 sickerin

Hi @sickerin , let me know how is it going or if you need any help.

rragundez avatar Feb 28 '24 15:02 rragundez