用于判断SoftMaxTiling结构是否符合基本块特征。
1 | bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2) |
1 | bool IsBasicBlockInSoftMax(SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2) |
接口 |
输入/输出 |
功能 |
|---|---|---|
tiling |
输入 |
待判断的SoftMaxTiling结构,支持optiling::SoftMaxTiling形式入参和SoftMaxTiling形式入参。 |
dataTypeSize |
输入 |
参与计算的srcTensor的数据类型大小,比如half=2。 |
无