bugfix: fix prefill kernel uris for aot compilation

[yzh119]

mask is no longer part of uris, this PR fixes the issue, otherwise our aot wheels will still trigger JIT compilation for prefill kernels.