已合并
megamoe fix ut #6689
lyt_claire创建于 6月10日
megamoe fix ut #6689
已合并
共 2 个文件变更+4-2
| @@ -58,10 +58,10 @@ TEST_F(MegaMoeArch22TilingTest, Test0) | |||
| 58 | inputs.push_back({{{kNHalf, kHiddenSize}, {kNHalf, kHiddenSize}}, ge::DT_INT4, ge::FORMAT_ND}); | 58 | inputs.push_back({{{kNHalf, kHiddenSize}, {kNHalf, kHiddenSize}}, ge::DT_INT4, ge::FORMAT_ND}); |
| 59 | } | 59 | } |
| 60 | for (uint32_t i = 0; i < kExpertPerRank; i++) { | 60 | for (uint32_t i = 0; i < kExpertPerRank; i++) { |
| 61 | - inputs.push_back({{{kN}, {kN}}, ge::DT_FLOAT, ge::FORMAT_ND}); | 61 | + inputs.push_back({{{kN}, {kN}}, ge::DT_UINT64, ge::FORMAT_ND}); |
| 62 | } | 62 | } |
| 63 | for (uint32_t i = 0; i < kExpertPerRank; i++) { | 63 | for (uint32_t i = 0; i < kExpertPerRank; i++) { |
| 64 | - inputs.push_back({{{kHiddenSize}, {kHiddenSize}}, ge::DT_FLOAT, ge::FORMAT_ND}); | 64 | + inputs.push_back({{{kHiddenSize}, {kHiddenSize}}, ge::DT_UINT64, ge::FORMAT_ND}); |
| 65 | } | 65 | } |
| 66 | for (uint32_t i = 0; i < kExpertPerRank; i++) { | 66 | for (uint32_t i = 0; i < kExpertPerRank; i++) { |
| 67 | inputs.push_back({{{kN}, {kN}}, ge::DT_FLOAT, ge::FORMAT_ND}); | 67 | inputs.push_back({{{kN}, {kN}}, ge::DT_FLOAT, ge::FORMAT_ND}); |
| @@ -78,6 +78,8 @@ ascend910b: | |||
| 78 | - mc2/quant_all_reduce/tests/ut/op_host | 78 | - mc2/quant_all_reduce/tests/ut/op_host |
| 79 | - mc2/quant_reduce_scatter/op_host | 79 | - mc2/quant_reduce_scatter/op_host |
| 80 | - mc2/quant_reduce_scatter/tests/ut/op_host | 80 | - mc2/quant_reduce_scatter/tests/ut/op_host |
| 81 | + - mc2/mega_moe/op_host | ||
| 82 | + - mc2/mega_moe/tests/ut/op_host | ||
| 81 | options: | 83 | options: |
| 82 | - ascend910b | 84 | - ascend910b |
| 83 | 85 | ||