torch implementation for butterfly transform

HazyResearch / butterfly

Butterfly matrix multiplication in PyTorch

Apache License 2.0

164 stars 31 forks source link

torch implementation for butterfly transform #19

Open dorooddorood606 opened 3 years ago

dorooddorood606 commented 3 years ago

Hi could you please guide me how I can use the torch buttefly implementation on CUDA? and where the implementation are located? I see codes in butterfly and csrc folder, could you tell me which one is correct one? Any examples I can start from? @tridao thanks

dorooddorood606 commented 3 years ago

I am also looking for fast walsh transform in cuda, could you please guide me if there are any implementation available in the repo? I saw one but I was thinking this is in torch not cuda. thanks