Roberto L. CastroandCodex
6ddbab03de
Signed-off-by: LopezCastroRoberto <rocastro@redhat.com> Co-authored-by: Codex <codex@openai.com>
11 lines
307 B
Python
11 lines
307 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""Ops shared across model implementations."""
|
|
|
|
from .fused_qk_rmsnorm import _FUSED_Q_KV_RMSNORM_KERNEL, fused_q_kv_rmsnorm
|
|
|
|
__all__ = [
|
|
"_FUSED_Q_KV_RMSNORM_KERNEL",
|
|
"fused_q_kv_rmsnorm",
|
|
]
|