已合并
[Reduce模板] Reduce模板算子ASCENDC_TPL_SEL修改 #2460
Sunee创建于 3月9日
[Reduce模板] Reduce模板算子ASCENDC_TPL_SEL修改 #2460
已合并
共 7 个文件变更+30-59
| @@ -24,7 +24,7 @@ else() | |||
| 24 | FetchContent_Declare( | 24 | FetchContent_Declare( |
| 25 | opbase | 25 | opbase |
| 26 | GIT_REPOSITORY https://gitcode.com/cann/opbase.git | 26 | GIT_REPOSITORY https://gitcode.com/cann/opbase.git |
| 27 | - GIT_TAG 07383cb90f23b3c4e9b1e16eff40a048c888555f | 27 | + GIT_TAG c8d83f3e57a63a7375e89a2d6937452c0ae2e522 |
| 28 | GIT_PROGRESS TRUE | 28 | GIT_PROGRESS TRUE |
| 29 | SOURCE_DIR ${CANN_3RD_LIB_PATH}/opbase) | 29 | SOURCE_DIR ${CANN_3RD_LIB_PATH}/opbase) |
| 30 | 30 | ||
| @@ -234,8 +234,9 @@ ge::graphStatus RIGDavidTilingImpl::DoTiling(gert::TilingContext* context) | |||
| 234 | 234 | ||
| 235 | context_->SetBlockDim(realCoreNum_); | 235 | context_->SetBlockDim(realCoreNum_); |
| 236 | 236 | ||
| 237 | - const uint64_t tilingKey = GET_TPL_TILING_KEY(0, 0, 0, templateNum_); | 237 | + repeatInterleaveGradTilingKey key; |
| 238 | - | 238 | + uint64_t tilingKey; |
| 239 | + GEN_REDUCE_TILING_KEY(tilingKey, key.ReduceTiling, templateNum_); | ||
| 239 | context_->SetTilingKey(tilingKey); | 240 | context_->SetTilingKey(tilingKey); |
| 240 | size_t* workspaces = context_->GetWorkspaceSizes(1); | 241 | size_t* workspaces = context_->GetWorkspaceSizes(1); |
| 241 | OP_CHECK_NULL_WITH_CONTEXT(context_, workspaces); | 242 | OP_CHECK_NULL_WITH_CONTEXT(context_, workspaces); |
| @@ -29,25 +29,25 @@ ASCENDC_TPL_SEL( | |||
| 29 | // empty | 29 | // empty |
| 30 | ASCENDC_TPL_ARGS_SEL( | 30 | ASCENDC_TPL_ARGS_SEL( |
| 31 | REDUCE_TPL_KEY_SEL_EMPTY(), | 31 | REDUCE_TPL_KEY_SEL_EMPTY(), |
| 32 | - ASCENDC_TPL_UINT_DECL(TemplateNum, REPEAT_INTERLEAVE_GRAD_BIT_WIDTH, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), | 32 | + ASCENDC_TPL_UINT_SEL(TemplateNum, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), |
| 33 | // A | 33 | // A |
| 34 | ASCENDC_TPL_ARGS_SEL( | 34 | ASCENDC_TPL_ARGS_SEL( |
| 35 | REDUCE_TPL_KEY_SEL_A(), | 35 | REDUCE_TPL_KEY_SEL_A(), |
| 36 | - ASCENDC_TPL_UINT_DECL(TemplateNum, REPEAT_INTERLEAVE_GRAD_BIT_WIDTH, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), | 36 | + ASCENDC_TPL_UINT_SEL(TemplateNum, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), |
| 37 | // AR | 37 | // AR |
| 38 | ASCENDC_TPL_ARGS_SEL( | 38 | ASCENDC_TPL_ARGS_SEL( |
| 39 | REDUCE_TPL_KEY_SEL_AR_NORMAL(), | 39 | REDUCE_TPL_KEY_SEL_AR_NORMAL(), |
| 40 | - ASCENDC_TPL_UINT_DECL(TemplateNum, REPEAT_INTERLEAVE_GRAD_BIT_WIDTH, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), | 40 | + ASCENDC_TPL_UINT_SEL(TemplateNum, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), |
| 41 | ASCENDC_TPL_ARGS_SEL( | 41 | ASCENDC_TPL_ARGS_SEL( |
| 42 | REDUCE_TPL_KEY_SEL_AR_GROUP(), | 42 | REDUCE_TPL_KEY_SEL_AR_GROUP(), |
| 43 | - ASCENDC_TPL_UINT_DECL(TemplateNum, REPEAT_INTERLEAVE_GRAD_BIT_WIDTH, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), | 43 | + ASCENDC_TPL_UINT_SEL(TemplateNum, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), |
| 44 | 44 | ||
| 45 | // ARA | 45 | // ARA |
| 46 | ASCENDC_TPL_ARGS_SEL( | 46 | ASCENDC_TPL_ARGS_SEL( |
| 47 | REDUCE_TPL_KEY_SEL_ARA_NORMAL(), | 47 | REDUCE_TPL_KEY_SEL_ARA_NORMAL(), |
| 48 | - ASCENDC_TPL_UINT_DECL(TemplateNum, REPEAT_INTERLEAVE_GRAD_BIT_WIDTH, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), | 48 | + ASCENDC_TPL_UINT_SEL(TemplateNum, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), |
| 49 | ASCENDC_TPL_ARGS_SEL( | 49 | ASCENDC_TPL_ARGS_SEL( |
| 50 | REDUCE_TPL_KEY_SEL_ARA_GROUP(), | 50 | REDUCE_TPL_KEY_SEL_ARA_GROUP(), |
| 51 | - ASCENDC_TPL_UINT_DECL(TemplateNum, REPEAT_INTERLEAVE_GRAD_BIT_WIDTH, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), ); | 51 | + ASCENDC_TPL_UINT_SEL(TemplateNum, ASCENDC_TPL_UI_RANGE, 1, 0, 4)), ); |
| 52 | 52 | ||
| 53 | 53 | ||
| @@ -19,7 +19,6 @@ | |||
| 19 | 19 | ||
| 20 | 20 | ||
| 21 | 21 | ||
| 22 | - | ||
| 23 | 22 | ||
| 24 | 23 | ||
| 25 | 24 | ||
| @@ -37,28 +36,19 @@ ASCENDC_TPL_ARGS_DECL(BinaryCrossEntropy, REDUCE_TPL_KEY_DECL(), | |||
| 37 | 36 | ||
| 38 | ASCENDC_TPL_SEL( | 37 | ASCENDC_TPL_SEL( |
| 39 | ASCENDC_TPL_ARGS_SEL( | 38 | ASCENDC_TPL_ARGS_SEL( |
| 40 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 39 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 41 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, BCE_TPL_SCH_MODE_0), | ||
| 42 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 43 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 44 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, BCE_NONE), | 40 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, BCE_NONE), |
| 45 | ASCENDC_TPL_UINT_SEL(HasWeight, ASCENDC_TPL_UI_LIST, 0, BCE_HAS_WEIGHT), | 41 | ASCENDC_TPL_UINT_SEL(HasWeight, ASCENDC_TPL_UI_LIST, 0, BCE_HAS_WEIGHT), |
| 46 | ASCENDC_TPL_UINT_SEL(DType, ASCENDC_TPL_UI_LIST, BCE_TPL_FP16) | 42 | ASCENDC_TPL_UINT_SEL(DType, ASCENDC_TPL_UI_LIST, BCE_TPL_FP16) |
| 47 | ), | 43 | ), |
| 48 | ASCENDC_TPL_ARGS_SEL( | 44 | ASCENDC_TPL_ARGS_SEL( |
| 49 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 45 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 50 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, BCE_TPL_SCH_MODE_0), | ||
| 51 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 52 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 53 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, BCE_NONE), | 46 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, BCE_NONE), |
| 54 | ASCENDC_TPL_UINT_SEL(HasWeight, ASCENDC_TPL_UI_LIST, 0, BCE_HAS_WEIGHT), | 47 | ASCENDC_TPL_UINT_SEL(HasWeight, ASCENDC_TPL_UI_LIST, 0, BCE_HAS_WEIGHT), |
| 55 | ASCENDC_TPL_UINT_SEL(DType, ASCENDC_TPL_UI_LIST, BCE_TPL_BF16) | 48 | ASCENDC_TPL_UINT_SEL(DType, ASCENDC_TPL_UI_LIST, BCE_TPL_BF16) |
| 56 | ), | 49 | ), |
| 57 | ASCENDC_TPL_ARGS_SEL( | 50 | ASCENDC_TPL_ARGS_SEL( |
| 58 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 51 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 59 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, BCE_TPL_SCH_MODE_0), | ||
| 60 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 61 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 62 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, BCE_NONE), | 52 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, BCE_NONE), |
| 63 | ASCENDC_TPL_UINT_SEL(HasWeight, ASCENDC_TPL_UI_LIST, 0, BCE_HAS_WEIGHT), | 53 | ASCENDC_TPL_UINT_SEL(HasWeight, ASCENDC_TPL_UI_LIST, 0, BCE_HAS_WEIGHT), |
| 64 | ASCENDC_TPL_UINT_SEL(DType, ASCENDC_TPL_UI_LIST, BCE_TPL_FP32) | 54 | ASCENDC_TPL_UINT_SEL(DType, ASCENDC_TPL_UI_LIST, BCE_TPL_FP32) |
| @@ -29,26 +29,17 @@ ASCENDC_TPL_ARGS_DECL(LpLoss, REDUCE_TPL_KEY_DECL(), | |||
| 29 | 29 | ||
| 30 | ASCENDC_TPL_SEL( | 30 | ASCENDC_TPL_SEL( |
| 31 | ASCENDC_TPL_ARGS_SEL( | 31 | ASCENDC_TPL_ARGS_SEL( |
| 32 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 32 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 33 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | ||
| 34 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 35 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 36 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 33 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 37 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, LPLOSS_TPL_FP16)), | 34 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, LPLOSS_TPL_FP16)), |
| 38 | 35 | ||
| 39 | ASCENDC_TPL_ARGS_SEL( | 36 | ASCENDC_TPL_ARGS_SEL( |
| 40 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 37 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 41 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | ||
| 42 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 43 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 44 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 38 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 45 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, LPLOSS_TPL_FP32)), | 39 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, LPLOSS_TPL_FP32)), |
| 46 | 40 | ||
| 47 | ASCENDC_TPL_ARGS_SEL( | 41 | ASCENDC_TPL_ARGS_SEL( |
| 48 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 42 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 49 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | ||
| 50 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 51 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 52 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 43 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 53 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, LPLOSS_TPL_BFP16)), | 44 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, LPLOSS_TPL_BFP16)), |
| 54 | 45 | ||
| @@ -32,39 +32,37 @@ ASCENDC_TPL_ARGS_DECL( | |||
| 32 | 32 | ||
| 33 | ASCENDC_TPL_SEL( | 33 | ASCENDC_TPL_SEL( |
| 34 | ASCENDC_TPL_ARGS_SEL( | 34 | ASCENDC_TPL_ARGS_SEL( |
| 35 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | 35 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 36 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 37 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 38 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 36 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 39 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP16)), | 37 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP16)), |
| 40 | 38 | ||
| 41 | ASCENDC_TPL_ARGS_SEL( | 39 | ASCENDC_TPL_ARGS_SEL( |
| 42 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | 40 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 43 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 44 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 45 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 41 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 46 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), | 42 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), |
| 47 | 43 | ||
| 48 | ASCENDC_TPL_ARGS_SEL( | 44 | ASCENDC_TPL_ARGS_SEL( |
| 49 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | 45 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 50 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 51 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 52 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 46 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 53 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_BFP16)), | 47 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_BFP16)), |
| 54 | 48 | ||
| 55 | ASCENDC_TPL_ARGS_SEL( | 49 | ASCENDC_TPL_ARGS_SEL( |
| 56 | - REDUCE_TPL_KEY_SEL_EMPTY(), ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | 50 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 51 | + ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | ||
| 57 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), | 52 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), |
| 58 | 53 | ||
| 59 | ASCENDC_TPL_ARGS_SEL( | 54 | ASCENDC_TPL_ARGS_SEL( |
| 60 | - REDUCE_TPL_KEY_SEL_AR_NORMAL(), ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | 55 | + REDUCE_TPL_KEY_SEL_AR_NORMAL(), |
| 56 | + ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | ||
| 61 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), | 57 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), |
| 62 | 58 | ||
| 63 | ASCENDC_TPL_ARGS_SEL( | 59 | ASCENDC_TPL_ARGS_SEL( |
| 64 | - REDUCE_TPL_KEY_SEL_AR_GROUP(), ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | 60 | + REDUCE_TPL_KEY_SEL_AR_GROUP(), |
| 61 | + ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | ||
| 65 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), | 62 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32)), |
| 66 | 63 | ||
| 67 | ASCENDC_TPL_ARGS_SEL( | 64 | ASCENDC_TPL_ARGS_SEL( |
| 68 | - REDUCE_TPL_KEY_SEL_A(), ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | 65 | + REDUCE_TPL_KEY_SEL_A(), |
| 66 | + ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, MSELOSS_SUM, MSELOSS_MEAN), | ||
| 69 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32))); | 67 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, MSELOSS_TPL_FP32))); |
| 70 | 68 | ||
| @@ -29,26 +29,17 @@ ASCENDC_TPL_ARGS_DECL(SMOOTHL1LOSSV2, REDUCE_TPL_KEY_DECL(), | |||
| 29 | 29 | ||
| 30 | ASCENDC_TPL_SEL( | 30 | ASCENDC_TPL_SEL( |
| 31 | ASCENDC_TPL_ARGS_SEL( | 31 | ASCENDC_TPL_ARGS_SEL( |
| 32 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 32 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 33 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | ||
| 34 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 35 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 36 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 33 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 37 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, SMOOTHL1LOSSV2_TPL_FP16)), | 34 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, SMOOTHL1LOSSV2_TPL_FP16)), |
| 38 | 35 | ||
| 39 | ASCENDC_TPL_ARGS_SEL( | 36 | ASCENDC_TPL_ARGS_SEL( |
| 40 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 37 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 41 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | ||
| 42 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 43 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 44 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 38 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 45 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, SMOOTHL1LOSSV2_TPL_FP32)), | 39 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, SMOOTHL1LOSSV2_TPL_FP32)), |
| 46 | 40 | ||
| 47 | ASCENDC_TPL_ARGS_SEL( | 41 | ASCENDC_TPL_ARGS_SEL( |
| 48 | - ASCENDC_TPL_KERNEL_TYPE_SEL(ASCENDC_TPL_AIV_ONLY), | 42 | + REDUCE_TPL_KEY_SEL_EMPTY(), |
| 49 | - ASCENDC_TPL_UINT_SEL(PatternID, ASCENDC_TPL_UI_LIST, 0), | ||
| 50 | - ASCENDC_TPL_UINT_SEL(LoopARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 51 | - ASCENDC_TPL_UINT_SEL(LoopInnerARCount, ASCENDC_TPL_UI_LIST, 0), | ||
| 52 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), | 43 | ASCENDC_TPL_UINT_SEL(Reduction, ASCENDC_TPL_UI_LIST, 0), |
| 53 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, SMOOTHL1LOSSV2_TPL_BFP16)), | 44 | ASCENDC_TPL_UINT_SEL(Dtype, ASCENDC_TPL_UI_LIST, SMOOTHL1LOSSV2_TPL_BFP16)), |
| 54 | 45 | ||