diff options
Diffstat (limited to 'eval/src/tests/tensor/dense_inplace_join_function/dense_inplace_join_function_test.cpp')
-rw-r--r-- | eval/src/tests/tensor/dense_inplace_join_function/dense_inplace_join_function_test.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/eval/src/tests/tensor/dense_inplace_join_function/dense_inplace_join_function_test.cpp b/eval/src/tests/tensor/dense_inplace_join_function/dense_inplace_join_function_test.cpp index c794b81f573..daa84bd226f 100644 --- a/eval/src/tests/tensor/dense_inplace_join_function/dense_inplace_join_function_test.cpp +++ b/eval/src/tests/tensor/dense_inplace_join_function/dense_inplace_join_function_test.cpp @@ -38,6 +38,7 @@ EvalFixture::ParamRepo make_params() { .add("con_x5_A", spec({x(5)}, seq)) .add("con_x5_B", spec({x(5)}, seq)) .add("con_x5_C", spec({x(5)}, seq)) + .add("con_y3_A", spec({y(3)}, seq)) .add("con_x5y3_A", spec({x(5),y(3)}, seq)) .add("con_x5y3_B", spec({x(5),y(3)}, seq)) .add_mutable("mut_dbl_A", spec(1.5)) @@ -144,4 +145,8 @@ TEST("require that mapped tensors are not optimized") { TEST_DO(verify_not_optimized("mut_x_sparse+mut_x_sparse")); } +TEST("require that output from xw product can be optimized") { + TEST_DO(verify_optimized("reduce(con_x5_A*con_x5y3_B,sum,x)+con_y3_A", 1, -1)); +} + TEST_MAIN() { TEST_RUN_ALL(); } |