Optimize prepare_lens by removing device transfer#289

iforgetmyname merged 2 commits intosgl-project:mainsgl-project/sgl-kernel-npu:mainfrom

shengzhaotian:triton_fla_cpuopshengzhaotian/sgl-kernel-npu:triton_fla_cpuopCopy head branch name to clipboard