[CI] Remove unused imports with Ruff to pre-commit config, only to benchmarks/docs/examples folder (#3969)

benchmark/kernels/decoding_attention_triton/triton_flashinfer_cudnn.py

@@ -4,8 +4,6 @@ import math
 import cudnn
 import torch
 import torch.utils.benchmark as benchmark
-import triton
-import triton.language as tl
 from flashinfer import BatchDecodeWithPagedKVCacheWrapper
 
 from sglang.srt.layers.attention.triton_ops.decode_attention import decode_attention_fwd

benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_decode.py

@@ -1,6 +1,5 @@
 import itertools
 import math
-import os
 from typing import Optional, Tuple
 
 import torch

benchmark/kernels/rmsnorm/benchmark_rmsnorm.py

@@ -3,7 +3,6 @@ from typing import Optional, Tuple, Union
 
 import torch
 import triton
-import triton.language as tl
 from flashinfer.norm import fused_add_rmsnorm, rmsnorm
 from torch import nn
 from vllm import _custom_ops as vllm_ops

benchmark/kernels/scheduler_batch/benchmark_write_req_to_token_pool_triton.py

@@ -1,9 +1,6 @@
 import itertools
 import os
-from typing import List
 
-import numpy as np
-import pytest
 import torch
 import triton
 import triton.language as tl