Skip to content

Commit

Permalink
[Fix](mlu-ops):Fix mutual information forward case error (#1126)
Browse files Browse the repository at this point in the history
Co-authored-by: wangrt1 <“[email protected]”>
  • Loading branch information
wangrt1 and wangrt1 authored Oct 24, 2024
1 parent bbbc4fd commit 9db19ff
Showing 1 changed file with 0 additions and 16 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -262,22 +262,6 @@ INSTANTIATE_TEST_CASE_P(
testing::Values(MLUOP_UNKNOWN_DEVICE),
testing::Values(MLUOP_STATUS_SUCCESS)));

INSTANTIATE_TEST_CASE_P(
zero_element_2, mutual_information_forward_general,
testing::Combine(
testing::Values(MLUOpTensorParam{MLUOP_LAYOUT_ARRAY, MLUOP_DTYPE_FLOAT,
3, std::vector<int>({4, 0, 1})}),
testing::Values(MLUOpTensorParam{MLUOP_LAYOUT_ARRAY, MLUOP_DTYPE_FLOAT,
3, std::vector<int>({4, 1, 0})}),
testing::Values(MLUOpTensorParam{MLUOP_LAYOUT_ARRAY, MLUOP_DTYPE_INT64,
2, std::vector<int>({4, 4})}),
testing::Values(MLUOpTensorParam{MLUOP_LAYOUT_ARRAY, MLUOP_DTYPE_FLOAT,
3, std::vector<int>({4, 1, 1})}),
testing::Values(MLUOpTensorParam{MLUOP_LAYOUT_ARRAY, MLUOP_DTYPE_FLOAT,
1, std::vector<int>({4})}),
testing::Values(MLUOP_UNKNOWN_DEVICE),
testing::Values(MLUOP_STATUS_SUCCESS)));

INSTANTIATE_TEST_CASE_P(
py_error_dim, mutual_information_forward_general,
testing::Combine(
Expand Down

0 comments on commit 9db19ff

Please sign in to comment.