bamos / block

An intelligent block matrix library for numpy, PyTorch, and beyond.
Apache License 2.0
297 stars 33 forks source link

Request Support For the Newest Pytorch Version #19

Open rychong opened 4 years ago

rychong commented 4 years ago

This package pytorch_fft is very usefull and important to me. The fft of pytorch torch.fft ignore the batch dims, thus it run very slowly. While i need apply fft to many matrix, only this package can do this.

But when i install this package, it show that "torch.utils.ffi is deprecated. Please use cpp extensions instead". It can run in Pythorch 0.4, but it doesn't suppert the newest version of Pytorch, I hope you can fixed this problem, thus we can replace torch.fft with this package in the newest version of pytorch.