No module named 'xentropy_cuda_lib'

[zhuzn]

when I use from flash_attn.losses.cross_entropy import CrossEntropyLoss as FlashCrossEntropyLoss, I get a error, ModuleNotFoundError: No module named 'xentropy_cuda_lib. I don't find xentropy_cuda_lib that I use conda or pip. how do i install xentropy_cuda_lib？
Thanks

[tridao]

You can do cd csrc/xentropy && pip install ..
We have some of the instructions here but we'll eventually move them to the main README.

[zhuzn]

Thanks for your reply, I'll try testing it.