@@ -54,31 +54,22 @@ class OpDiagonalCopyOutTest : public ::testing::Test {
54
54
template <typename CTYPE, ScalarType DTYPE>
55
55
void run_2d_complex_dtype () {
56
56
TensorFactory<DTYPE> tf;
57
- constexpr auto REAL_DTYPE = executorch::runtime::toRealValueType (DTYPE);
58
- using REAL_CTYPE =
59
- typename executorch::runtime::ScalarTypeToCppType<REAL_DTYPE>::type;
60
57
Tensor input = tf.make (
61
58
{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 )});
82
73
op_diagonal_copy_out (input, 1 , 1 , 0 , out);
83
74
EXPECT_TENSOR_CLOSE (out, out_expected);
84
75
}
0 commit comments