使用说明

Ascend C提供一组SoftMax Tiling API,方便用户获取SoftMax kernel计算时所需的Tiling参数。

获取Tiling参数主要分为如下两步:

  1. 先获取SoftMax接口计算所需最小临时空间大小,用于合理分配计算空间。
  2. 获取输入SoftMax kernel侧接口所需tiling参数,需要传入输入shape,剩余的可供softmax接口计算的空间大小和计算的数据类型。

对于SoftMax/SimpleSoftMax请参考SoftMax/SimpleSoftMax Tiling接口;对于SoftmaxFlash请参考SoftmaxFlash Tiling接口