Skip to content

Commit d6e4c18

Browse files
manuelcandaleskeyprocedure
authored andcommitted
clean up complex tests
Differential Revision: D72197507 Pull Request resolved: pytorch#10213
1 parent 21c37aa commit d6e4c18

File tree

2 files changed

+15
-29
lines changed

2 files changed

+15
-29
lines changed

kernels/test/op_abs_test.cpp

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,7 @@ class OpAbsTest : public OperatorTest {
4444
TensorFactory<DTYPE> tf;
4545
constexpr auto REAL_DTYPE = executorch::runtime::toRealValueType(DTYPE);
4646
TensorFactory<REAL_DTYPE> tf_out;
47-
using REAL_CTYPE =
48-
typename executorch::runtime::ScalarTypeToCppType<REAL_DTYPE>::type;
49-
Tensor in = tf.make(
50-
{1, 2},
51-
{CTYPE{REAL_CTYPE(3), REAL_CTYPE(4)},
52-
CTYPE{REAL_CTYPE(5), REAL_CTYPE(12)}});
47+
Tensor in = tf.make({1, 2}, {CTYPE(3, 4), CTYPE(5, 12)});
5348
Tensor out = tf_out.zeros({1, 2});
5449
Tensor expected = tf_out.make({1, 2}, {5, 13});
5550
Tensor ret = op_abs_out(in, out);

kernels/test/op_diagonal_copy_test.cpp

Lines changed: 14 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -54,31 +54,22 @@ class OpDiagonalCopyOutTest : public ::testing::Test {
5454
template <typename CTYPE, ScalarType DTYPE>
5555
void run_2d_complex_dtype() {
5656
TensorFactory<DTYPE> tf;
57-
constexpr auto REAL_DTYPE = executorch::runtime::toRealValueType(DTYPE);
58-
using REAL_CTYPE =
59-
typename executorch::runtime::ScalarTypeToCppType<REAL_DTYPE>::type;
6057
Tensor input = tf.make(
6158
{3, 4},
62-
{CTYPE{REAL_CTYPE(1), REAL_CTYPE(1)},
63-
CTYPE{REAL_CTYPE(2), REAL_CTYPE(2)},
64-
CTYPE{REAL_CTYPE(3), REAL_CTYPE(3)},
65-
CTYPE{REAL_CTYPE(4), REAL_CTYPE(4)},
66-
CTYPE{REAL_CTYPE(5), REAL_CTYPE(5)},
67-
CTYPE{REAL_CTYPE(6), REAL_CTYPE(6)},
68-
CTYPE{REAL_CTYPE(7), REAL_CTYPE(7)},
69-
CTYPE{REAL_CTYPE(8), REAL_CTYPE(8)},
70-
CTYPE{REAL_CTYPE(9), REAL_CTYPE(9)},
71-
CTYPE{REAL_CTYPE(10), REAL_CTYPE(10)},
72-
CTYPE{REAL_CTYPE(11), REAL_CTYPE(11)},
73-
CTYPE{REAL_CTYPE(12), REAL_CTYPE(12)}});
74-
Tensor out = tf.make(
75-
{2},
76-
{CTYPE{REAL_CTYPE(0), REAL_CTYPE(0)},
77-
CTYPE{REAL_CTYPE(0), REAL_CTYPE(0)}});
78-
Tensor out_expected = tf.make(
79-
{2},
80-
{CTYPE{REAL_CTYPE(5), REAL_CTYPE(5)},
81-
CTYPE{REAL_CTYPE(10), REAL_CTYPE(10)}});
59+
{CTYPE(1, 1),
60+
CTYPE(2, 2),
61+
CTYPE(3, 3),
62+
CTYPE(4, 4),
63+
CTYPE(5, 5),
64+
CTYPE(6, 6),
65+
CTYPE(7, 7),
66+
CTYPE(8, 8),
67+
CTYPE(9, 9),
68+
CTYPE(10, 10),
69+
CTYPE(11, 11),
70+
CTYPE(12, 12)});
71+
Tensor out = tf.make({2}, {CTYPE(0, 0), CTYPE(0, 0)});
72+
Tensor out_expected = tf.make({2}, {CTYPE(5, 5), CTYPE(10, 10)});
8273
op_diagonal_copy_out(input, 1, 1, 0, out);
8374
EXPECT_TENSOR_CLOSE(out, out_expected);
8475
}

0 commit comments

Comments
 (0)