llama : add check for KV cache shifts (#10401)

llama : add check for KV cache shifts (#10401) #16828