@@ -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