[Bugfix][Logprobs] Fix logprobs op to support more backend (vllm-project#21591)

Signed-off-by: MengqingCao <[email protected]>

Author: MengqingCao

vllm/v1/sample/ops/logprobs.py

@@ -4,8 +4,10 @@
 
 import torch
 
+from vllm.platforms import current_platform
 
-@torch.compile(dynamic=True)
+
+@torch.compile(dynamic=True, backend=current_platform.simple_compile_backend)
 def batched_count_greater_than(x: torch.Tensor,
                                values: torch.Tensor) -> torch.Tensor:
     """