dfdx
dfdx copied to clipboard
Add complex numbers
Closes #821.
I had to use a proxy object to implement cuda traits. It built with cargo build -F complex,cuda,nightly,f16
How does this interact with cuda kernels?
I don't think it would work at all with cuda kernels atm. Not familiar with how cuda types work, can you overload operators for a custom complex struct?
I don't think it would work at all with cuda kernels atm. Not familiar with how cuda types work, can you overload operators for a custom complex struct?
We could potentially just merge this for Cpu and make sure we get compile errors for cuda.
You can overload operators in c++, but I'm not sure that's sufficient to make the kernels work in the same way as they do in rust land.