[llama] Update kv cache to have read/write functions (#280)

[llama] Update kv cache to have read/write functions (#280) #50