e9be5323f0
Signed-off-by: roi.pony <roi.pony@ibm.com> Co-authored-by: roi.pony <roi.pony@ibm.com> Co-authored-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>
8 lines
276 B
Python
8 lines
276 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""Fused Triton kernels for late-interaction (MaxSim) scoring."""
|
|
|
|
from .flash_maxsim_rerank import flash_maxsim_rerank_direct
|
|
|
|
__all__ = ["flash_maxsim_rerank_direct"]
|