12 lines
340 B
Python
12 lines
340 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""Ops shared across model implementations."""
|
|
|
|
from .fused_allreduce_rms_norm import fused_allreduce_rms_norm
|
|
from .fused_qk_rmsnorm import fused_q_kv_rmsnorm
|
|
|
|
__all__ = [
|
|
"fused_allreduce_rms_norm",
|
|
"fused_q_kv_rmsnorm",
|
|
]
|