Skip to content

IsBasicBlockInSoftMax

功能说明

用于判断SoftMaxTiling结构是否符合基本块特征。

函数原型

  • AscendC::optiling命名空间下的计算接口

    Text
    bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2)
    
  • AscendC命名空间下的计算接口

    Text
    bool IsBasicBlockInSoftMax(AscendC::tiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2)
    

参数说明

表1 参数列表

接口输入/输出功能
tiling输入待判断的SoftMaxTiling结构,支持optiling::SoftMaxTiling形式入参和AscendC::tiling::SoftMaxTiling形式入参。
dataTypeSize输入参与计算的srcTensor的数据类型大小,比如half=2。

返回值说明

  • 返回true表示SoftMaxTiling结构满足基本块Tiling特征。
  • 返回false表示SoftMaxTiling结构不满足基本块Tiling特征。

约束说明

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区