search

idiap / fast-transformers

Pytorch library for fast transformer implementations
1.65k stars 179 forks source link

ModuleNotFoundError: No module named 'aggregate.aggregate_cpu' #93

Closed TianhaoFu closed 3 years ago

TianhaoFu commented 3 years ago

Hi, when i want to use clusted attention, I came across a bug. It seems because the aggregate_cpu.cu is a cuda type file.

BTW， I insert the source code in code and i came across such bug.

Can you give me some advices on this bug?

Thanks:)

angeloskath commented 3 years ago

Hi, did you compile the library first? You can do this by running python setup.py build_ext --inplace. If you installed via pip then it should have automatically compiled the library first.

Cheers, Angelos

TianhaoFu commented 3 years ago

Hi, did you compile the library first? You can do this by running python setup.py build_ext --inplace. If you installed via pip then it should have automatically compiled the library first.

Cheers, Angelos

thanks. Its work