Skip to content

GitLab

Unverified Commit 72ad2735 authored by QiliangCui's avatar QiliangCui Committed by GitHub
Browse files

Remove torch_xla.tpu.version() from pallas.py. (#21065) 


Signed-off-by: default avatarQiliang Cui <derrhein@gmail.com>
parent 01513a33
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
vllm/v1/attention/backends/pallas.py
View file @ 72ad2735
......@@ -167,10 +167,6 @@ class PallasAttentionBackendImpl(AttentionImpl):
"are not implemented for "
"PallasAttentionBackendImpl")
tpu_version = torch_xla.tpu.version()
if tpu_version < 4:
raise NotImplementedError("TPU version must be 4 or higher.")
def forward(
self,
layer: AttentionLayer,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment