6 Commits

Author	SHA1	Message	Date
Chengji Yao	3a1d8940ae	[TPU] support fp8 kv cache quantization (#19292) ... Signed-off-by: Chengji Yao <chengjiyao@google.com>	2025-07-20 03:01:00 +00:00
Chengji Yao	eb58f5953d	[TPU][Bugfix] fix test_pallas (#20666) ... Signed-off-by: Chengji Yao <chengjiyao@google.com>	2025-07-09 09:32:48 -07:00
Chengji Yao	04e1642e32	[TPU] add kv cache update kernel (#19928) ... Signed-off-by: Chengji Yao <chengjiyao@google.com>	2025-06-26 10:01:37 -07:00
Simon Mo	02f0c7b220	[Misc] Add SPDX-FileCopyrightText (#19100) ... Signed-off-by: simon-mo <simon.mo@hey.com>	2025-06-03 11:20:17 -07:00
Chengji Yao	b1eb4ca152	[TPU] Update PyTorch/XLA (#16288) ... Signed-off-by: Chengji Yao <chengjiyao@google.com>	2025-04-09 14:46:32 +08:00
iefgnoix	b6be6f8d1e	[TPU] Support sliding window and logit soft capping in the paged attention kernel for TPU. (#15732) ... Signed-off-by: Xiongfei Wei <isaacwxf23@gmail.com>	2025-04-03 14:23:28 -07:00