---
title: TopKFusionPass
description: "该融合规则将TopK或TopKV2节点，基于平台的不同替换为TopKV2/TopKD/TopKV2D算子，或直接拆解为SegmentSort+MultiMerge的组合。详细场景如下。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/maintenref/graphubfusionref/atlasrr_30_0157.html
sourcePath: /source/zh/canncommercial/900/maintenref/graphubfusionref/atlasrr_30_0157.html
indexId: aab1ea6fb1309dfdf17a390bfb122e8ae5569986310b72221c9b7041c815174d78
---
# TopKFusionPass

#### 融合模式

该融合规则将TopK或TopKV2节点，基于平台的不同替换为TopKV2/TopKD/TopKV2D算子，或直接拆解为SegmentSort+MultiMerge的组合。详细场景如下。

场景一：TopK/TopKV2算子将被替换为TopKV2算子


场景二：TopK/TopKV2算子被替换为TopKD/TopKV2D算子


场景三：TopK/TopKV2算子将被拆分为SegmentSort+MultiMerge的组合


#### 使用约束

- 如下形态，不支持TopK算子的attr.sorted=false；当输入k非const tensor时，TopK/TopKV2算子会替换为TopKV2算子；当输入k是const tensor时，TopK/TopKV2算子会替换为TopKD/TopKV2D算子。

  - Atlas A3 训练系列产品 / Atlas A3 推理系列产品
  - Atlas A2 训练系列产品 / Atlas A2 推理系列产品
  - Atlas 200I/500 A2 推理产品
  - Atlas 推理系列产品
  - Atlas 训练系列产品
- Atlas 350 加速卡场景下，该融合规则不支持关闭。
- Atlas 350 加速卡场景下，TopK/TopKV2算子会替换为TopKV2算子。


#### 支持的型号

该Pass的有效性依赖于运行的产品是否支持相应算子类型TopK/TopKV2，具体信息请参考“Ascend IR算子规格说明(https://www.hiascend.comdocument/detail/zh/canncommercial/900/API/aolapi/operatorlist_00093.html)”章节。