构造函数
功能说明
用于创建一个Conv3DBackpropInput 单核Tiling对象。
函数原型
- 带参构造函数,需要传入硬件平台信息,推荐使用这类构造函数来获得更好的兼容性。
- 使用PlatformAscendC类传入信息
1
explicit Conv3DBpInputTiling(const platform_ascendc::PlatformAscendC& ascendcPlatform)
- 使用PlatformInfo传入信息
- 使用PlatformAscendC类传入信息
- 无参构造函数
1
Conv3DBpInputTiling()
- 基类构造函数
Conv3DBpInputTiling继承自基类Conv3DBpInputTilingBase,其构造函数如下:
1
Conv3DBpInputTilingBase()
1
explicit Conv3DBpInputTilingBase(const platform_ascendc::PlatformAscendC& ascendcPlatform)
1
explicit Conv3DBpInputTilingBase(const PlatformInfo& platform)
参数说明
参数名 |
输入/输出 |
描述 |
||
---|---|---|---|---|
ascendcPlatform |
输入 |
传入硬件平台的信息,PlatformAscendC定义请参见构造及析构函数。 |
||
platform |
输入 |
传入硬件版本以及AI Core中各个硬件单元提供的内存大小。PlatformInfo构造时通过构造及析构函数获取。 PlatformInfo结构定义如下,socVersion通过GetSocVersion获取并透传,各类硬件存储空间大小通过GetCoreMemSize获取并透传。
不推荐通过直接填值构造PlatformInfo的方式调用构造函数,例如PlatformInfo(, 1024, 1024, ..); |
在实现Host侧的Tiling函数时,platform_ascendc::PlatformAscendC用于获取一些硬件平台的信息,来支撑Tiling的计算,比如获取硬件平台的核数等信息。PlatformAscendC类提供获取这些平台信息的功能。
和platform_ascendc::PlatformAscendC不同的是,PlatformInfo则用于获取芯片版本以及AI Core中各个硬件单元提供的内存大小等只针对单个AI Core的信息。
约束说明
无
调用示例
- 无参构造函数
1 2 3 4 5 6
ConvBackpropApi::Conv3DBpInputTiling tiling; tiling.SetWeightType(ConvCommonApi::TPosition::GM,ConvCommonApi::ConvFormat::FRACTAL_Z_3D,ConvCommonApi::ConvDtype::FLOAT16); ... optiling::Conv3DBackpropInputTilingData tilingData; int ret = tiling.GetTiling(tilingData); // if ret = -1, gen tiling failed ...
- 带参构造函数
1 2 3 4 5 6
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); ConvBackpropApi::Conv3DBpInputTiling tiling(ascendcPlatform); tiling.SetWeightType(ConvCommonApi::TPosition::GM,ConvCommonApi::ConvFormat::FRACTAL_Z_3D,ConvCommonApi::ConvDtype::FLOAT16); ... optiling::Conv3DBackpropInputTilingData tilingData; int ret = tiling.GetTiling(tilingData); // if ret = -1, gen tiling failed