generic protocol for compilation wrappers

[YouJiacheng]

Fix #9932 without breaking #9950

[YouJiacheng]

@froystig WDYT?

[YouJiacheng]

Emmm, it seems that type checking tool doesn't support ParamSpec

error: Argument 1 to "partial" has incompatible type "Callable[[Callable[P, T], DefaultNamedArg(Union[int, Iterable[int], None], 'static_argnums'), DefaultNamedArg(Union[str, Iterable[str], None], 'static_argnames'), DefaultNamedArg(Optional[Device], 'device'), DefaultNamedArg(Optional[str], 'backend'), DefaultNamedArg(Union[int, Iterable[int]], 'donate_argnums'), DefaultNamedArg(bool, 'inline')], Wrapped[P, T]]"; expected "Callable[..., Wrapped[P, T]]"  [arg-type]

Argument 4 to "__call__" of "Wrapped" has incompatible type "Sequence[int]"; expected "P.args"  [arg-type]

[YouJiacheng]

Ah, I find that it is a mypy bug: it doesn't handle partial with decorator correctly, while pylance can.
python/mypy#12593