Host侧接口，用于获取Acos接口能完成计算所需最小的临时空间大小，此空间为预留空间，即需要保证预留有足够的物理空间，用于执行计算。
inline uint32_t GetAcosMinTmpSize(const ge::Shape &srcShape, const uint32_t typeSize, const bool isReuseSource);
接口 |
输入/输出 |
功能 |
|---|---|---|
srcShape |
输入 |
输入的shape信息。 |
typeSize |
输入 |
算子输入的数据类型大小，单位为字节。比如算子输入的数据类型为half，此处应传入2，即sizeof(half)。 |
isReuseSource |
输入 |
是否允许修改源操作数。 |
返回Acos接口完成计算所需最小临时空间大小。
Atlas A2训练系列产品
完整的调用样例请参考更多样例。
// 输入shape信息为1024;算子输入的数据类型为half;不允许修改源操作数
std::vector<int64> shape_vec = {1024};
ge::Shape shape(shape_vec);
auto tmp_size = GetAcosMinTmpSize(shape, sizeof(half), false);
输出数据(tmp_size): 12288