Repository navigation
feat: complete retrieval encoder operator support across backends - #1017
Merged
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
为 BGE embedding 和 reranker 补齐跨硬件的算子执行能力,使 InfiniTensor 能通过 InfiniOps 的公开接口完成检索 encoder 和分类头计算。
同时修复接入过程中暴露的外部内存、stream 顺序、输入约束和工作区分配问题,避免上层推理引擎为不同设备重复实现这些处理。
Motivation
知识库检索与精排需要执行词嵌入、归一化、attention、GELU 和分类头。现有算子已经覆盖大部分计算,但在 ONNX 语义、部分设备的类型转换,以及 ATen 与外部 runtime 配合执行方面仍有缺口。
本 PR 将这些能力和修复放在 InfiniOps 中,让调用方复用统一接口。模型加载、动态形状、文档检索和业务接口继续由上层实现。
Closes #
Type of Change
feat— new feature / new operator / new platformfix— bug fixperf— performance improvement (no behavioral change)refactor— code restructuring without behavior changetest— adding or fixing tests onlydocs— documentation onlybuild/ci— build system or CI configurationchore— tooling, formatting, or other non-code changes!in the Conventional Commits prefix or aBREAKING CHANGE:footer)Platforms Affected
WITH_CPU)WITH_NVIDIA)WITH_ILUVATAR)WITH_METAX)WITH_CAMBRICON)WITH_MOORE)WITH_ASCEND)WITH_TORCH)Smoke Test Result
Test Results on Supported Platforms
Full `pytest` output (optional)
Benchmark / Performance Impact
Notes for Reviewers