下载
中文
注册

DropOutDoMaskFusionPass

融合模式

该融合规则替换DropOutDoMaskV3D算子为DropOutDoMask。

融合后DropOutDoMaskV3D算子替换为DropOutDoMask算子,DropOutDoMask算子可以走DSL分支通过Elewise模板实现,支持UB融合。

融合成

使用约束

在支持DSA模块时融合规则生效且必须打开,在需要DropOutDoMask算子进行UB融合时该融合规则必须打开。

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品