accelerated-scan
accelerated-scan copied to clipboard
Question: integration with complex numbers
Hi, I was wondering if there are any plans to make the cuda code be compatible with complex numbers? This would be particularly helpful given that triton does not currently support complex numbers.
Thank you
Hi @ekellbuch, I didn't plan to add them explicitly, however having them would enable LRU implementation! However it would still be useful to implement the same functionality in triton to have feature parity. Even if triton doesn't explicitly support complex numbers we could implement complex multiplication ourselves.