Giters

abhijangda / fastkron

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Replacing torch.kron with fastkron

RaulPPelaez opened this issue · comments

commented

Hi, thanks for your nice contribution!

Are there any plans to write a replacement for torch.kron?
Either in torch itself or providing a custom CUDA/C++ extension through this repo.

That would entail writing a backwards pass for fastkron as well.

commented

Hi Raul,

Yes, the plan is to provide a Python extension that does Kronecker Matrix-Matrix Multiplication. However, I think that requires a few months. I am not sure about the backward pass yet. I think that is more evolved.