[Llama] Use rocm ukernel when available + use num_layer for pkv.

[Llama] Use rocm ukernel when available + use num_layer for pkv. #571