Is MAGNet not supported for CPU?

[fibr1900]

Hello,

I was trying to run the magnet_demo.ipynb when I got this error:

smallkF is not supported because: max(query.shape[-1] != value.shape[-1]) > 32 device=cpu (supported: {'cuda'}) operator wasn't built - see python -m xformers.info for more info unsupported embed per head: 64

I am running on a CPU and I'm interpreting this as it is not possible to run MAGNet on a CPU, is that correct?