修复custom_op后端分发问题

vllm/model_executor/custom_op.py

@@ -184,6 +184,8 @@ class CustomOp(nn.Module):
             return self.maybe_compile(self.forward_native, enable=compile_native)
 
         if current_platform.is_rocm():
+            if self.__class__.name == "sparse_attn_indexer":
+                return self.forward_cuda
             return self.forward_cuda
         elif current_platform.is_cpu():
             return self.forward_cpu