已合并
fix experimental pkg fail #899
liulinxiang创建于 1月25日
fix experimental pkg fail #899
已合并
共 6 个文件变更+12-31
| @@ -61,26 +61,7 @@ static ge::graphStatus GetPlatformInfo(gert::TilingContext* context, uint64_t& u | |||
| 61 | return ge::GRAPH_SUCCESS; | 61 | return ge::GRAPH_SUCCESS; |
| 62 | } | 62 | } |
| 63 | 63 | ||
| 64 | -// 获取属性,shape信息 | 64 | +static ge::graphStatus GetWorkspaceSize(gert::TilingContext* context) |
| 65 | -ge::graphStatus GetShapeAttrsInfo(gert::TilingContext* context, int64_t& totalIdx, ge::DataType& dataType) | ||
| 66 | -{ | ||
| 67 | - // 获取输入shape信息 | ||
| 68 | - auto inputX = context->GetInputShape(0); | ||
| 69 | - OP_CHECK_NULL_WITH_CONTEXT(context, inputX); | ||
| 70 | - totalIdx = inputX->GetStorageShape().GetShapeSize(); | ||
| 71 | - // dtype校验 | ||
| 72 | - const std::set<ge::DataType> supportedDtype = {ge::DT_FLOAT, ge::DT_INT32, ge::DT_INT16, ge::DT_FLOAT16}; | ||
| 73 | - auto inputDesc = context->GetInputDesc(0); | ||
| 74 | - OP_CHECK_NULL_WITH_CONTEXT(context, inputDesc); | ||
| 75 | - dataType = inputDesc->GetDataType(); | ||
| 76 | - if (supportedDtype.count(dataType) == 0) { | ||
| 77 | - OP_LOGE(context, "invalid dtype"); | ||
| 78 | - return ge::GRAPH_FAILED; | ||
| 79 | - } | ||
| 80 | - return ge::GRAPH_SUCCESS; | ||
| 81 | -} | ||
| 82 | - | ||
| 83 | -ge::graphStatus GetWorkspaceSize(gert::TilingContext* context) | ||
| 84 | { | 65 | { |
| 85 | auto ascendcPlatform = platform_ascendc:: PlatformAscendC(context->GetPlatformInfo()); | 66 | auto ascendcPlatform = platform_ascendc:: PlatformAscendC(context->GetPlatformInfo()); |
| 86 | uint32_t sysWorkspaceSize = ascendcPlatform.GetLibApiWorkSpaceSize(); | 67 | uint32_t sysWorkspaceSize = ascendcPlatform.GetLibApiWorkSpaceSize(); |
| @@ -14,10 +14,10 @@ | |||
| 14 | */ | 14 | */ |
| 15 | 15 | ||
| 16 | 16 | ||
| 17 | -#include "tiling_base/tiling_util.h" | 17 | +#include "op_host/tiling_util.h" |
| 18 | 18 | ||
| 19 | 19 | ||
| 20 | -#include "tiling_base/tiling_templates_registry.h" | 20 | +#include "op_host/tiling_templates_registry.h" |
| 21 | 21 | ||
| 22 | 22 | ||
| 23 | 23 | ||
| @@ -65,7 +65,7 @@ static ge::graphStatus GetPlatformInfo(gert::TilingContext* context, PlatformInf | |||
| 65 | } | 65 | } |
| 66 | 66 | ||
| 67 | // 获取属性,shape信息 | 67 | // 获取属性,shape信息 |
| 68 | -ge::graphStatus GetShapeAttrsInfo(gert::TilingContext* context, int64_t& totalIdx, ge::DataType& dataType) | 68 | +static ge::graphStatus GetShapeAttrsInfo(gert::TilingContext* context, int64_t& totalIdx, ge::DataType& dataType) |
| 69 | { | 69 | { |
| 70 | // 获取输入shape信息 | 70 | // 获取输入shape信息 |
| 71 | auto inputX = context->GetInputShape(0); | 71 | auto inputX = context->GetInputShape(0); |
| @@ -14,10 +14,10 @@ | |||
| 14 | */ | 14 | */ |
| 15 | 15 | ||
| 16 | 16 | ||
| 17 | -#include "tiling_base/tiling_util.h" | 17 | +#include "op_host/tiling_util.h" |
| 18 | 18 | ||
| 19 | 19 | ||
| 20 | -#include "tiling_base/tiling_templates_registry.h" | 20 | +#include "op_host/tiling_templates_registry.h" |
| 21 | 21 | ||
| 22 | 22 | ||
| 23 | 23 | ||
| @@ -25,8 +25,8 @@ | |||
| 25 | 25 | ||
| 26 | 26 | ||
| 27 | 27 | ||
| 28 | -#include "tiling_base/tiling_util.h" | 28 | +#include "op_host/tiling_util.h" |
| 29 | -#include "tiling_base/tiling_templates_registry.h" | 29 | +#include "op_host/tiling_templates_registry.h" |
| 30 | 30 | ||
| 31 | 31 | ||
| 32 | 32 | ||
| @@ -54,7 +54,7 @@ static ge::graphStatus GetPlatformInfo(gert::TilingContext* context, uint64_t& u | |||
| 54 | } | 54 | } |
| 55 | 55 | ||
| 56 | // 获取属性,shape信息 | 56 | // 获取属性,shape信息 |
| 57 | -ge::graphStatus GetShapeAttrsInfo(gert::TilingContext* context, int64_t& totalIdx, ge::DataType& dataType) | 57 | +static ge::graphStatus GetShapeAttrsInfo(gert::TilingContext* context, int64_t& totalIdx, ge::DataType& dataType) |
| 58 | { | 58 | { |
| 59 | // 获取输入shape信息 | 59 | // 获取输入shape信息 |
| 60 | auto inputX = context->GetInputShape(0); | 60 | auto inputX = context->GetInputShape(0); |
| @@ -72,7 +72,7 @@ ge::graphStatus GetShapeAttrsInfo(gert::TilingContext* context, int64_t& totalId | |||
| 72 | return ge::GRAPH_SUCCESS; | 72 | return ge::GRAPH_SUCCESS; |
| 73 | } | 73 | } |
| 74 | 74 | ||
| 75 | -ge::graphStatus GetWorkspaceSize(gert::TilingContext* context) | 75 | +static ge::graphStatus GetWorkspaceSize(gert::TilingContext* context) |
| 76 | { | 76 | { |
| 77 | auto ascendcPlatform = platform_ascendc:: PlatformAscendC(context->GetPlatformInfo()); | 77 | auto ascendcPlatform = platform_ascendc:: PlatformAscendC(context->GetPlatformInfo()); |
| 78 | uint32_t sysWorkspaceSize = ascendcPlatform.GetLibApiWorkSpaceSize(); | 78 | uint32_t sysWorkspaceSize = ascendcPlatform.GetLibApiWorkSpaceSize(); |