[BUG] missing symbol `make_causal`

Thanks for your great work!

I try to use linear attention kernel in TK. But encountered the error: missing symbol `make_causal`.

I have checked the code repo and there is no such a device function/template. Could you please help me to solve this problem?🫶