| 1 | # SPDX-License-Identifier: Apache-2.0 |
| 2 | # SPDX-FileCopyrightText: Copyright contributors to the vLLM project |
| 3 | |
| 4 | from vllm.model_executor.layers.hpc.gated_mla import ( |
| 5 | hpc_gated_mla_gemm, |
| 6 | hpc_gated_mla_supported, |
| 7 | ) |
| 8 | from vllm.model_executor.layers.hpc.hpc_ihc import ( |
| 9 | HpcIHCHead, |
| 10 | HpcIHCPost, |
| 11 | HpcIHCPre, |
| 12 | ) |
| 13 | from vllm.model_executor.layers.hpc.hpc_module import HpcModule |
| 14 | from vllm.model_executor.layers.hpc.rope_norm import HpcRopeNorm, QkNormPolicy |
| 15 | |
| 16 | __all__ = [ |
| 17 | "HpcIHCHead", |
| 18 | "HpcIHCPost", |
| 19 | "HpcIHCPre", |
| 20 | "HpcModule", |
| 21 | "HpcRopeNorm", |
| 22 | "QkNormPolicy", |
| 23 | "hpc_gated_mla_gemm", |
| 24 | "hpc_gated_mla_supported", |
| 25 | ] |