Hacker News new | ask | show | jobs
by ca_parody 2173 days ago
https://github.com/mratsim/Arraymancer probably has a lot of examples in it's implementation
1 comments

Yup. That's what popped up when I searched. Thanks for the reference though.
The CUDA kernels are stored here:

- https://github.com/mratsim/Arraymancer/blob/master/src/tenso...

With the higher order function:

- https://github.com/mratsim/Arraymancer/blob/master/src/tenso...

And their code generation:

- https://github.com/mratsim/Arraymancer/blob/master/src/tenso...

It's possible to avoid inlining C++ like they do here:

- https://github.com/jcosborn/cudanim/blob/338be782/cuda.nim#L...

but I will explore that later.

Thanks.