aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp
diff options
context:
space:
mode:
authorHåvard Pettersen <havardpe@yahooinc.com>2022-06-17 08:37:47 +0000
committerHåvard Pettersen <havardpe@yahooinc.com>2022-06-17 10:22:30 +0000
commit7fd09d198ff274109e4e95b0feda4662199fd723 (patch)
tree26d815aff6ff1d8ae1fb9d47bc05ad835230cd12 /eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp
parenta18e8e12faec6318c1d5768fd4edf42efd0ecb78 (diff)
optimize singledim sparse lookup
Diffstat (limited to 'eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp')
-rw-r--r--eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp14
1 files changed, 7 insertions, 7 deletions
diff --git a/eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp b/eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp
index 11654a38342..5cd63ce0492 100644
--- a/eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp
+++ b/eval/src/tests/instruction/dense_tensor_peek_function/dense_tensor_peek_function_test.cpp
@@ -56,10 +56,10 @@ TEST("require that tensor peek can be optimized for dense tensors") {
TEST_DO(verify("x3y2f{x:(a-1),y:(b)}", 0.0, 1, 0));
}
-TEST("require that tensor peek is not optimized for sparse tensor") {
+TEST("require that tensor peek is optimized differently for sparse tensor") {
TEST_DO(verify("xm{x:1}", 1.0, 0, 1));
- TEST_DO(verify("xm{x:(c)}", 3.0, 0, 1));
- TEST_DO(verify("xm{x:(c+1)}", 0.0, 0, 1));
+ TEST_DO(verify("xm{x:(c)}", 3.0, 0, 0));
+ TEST_DO(verify("xm{x:(c+1)}", 0.0, 0, 0));
}
TEST("require that tensor peek is not optimized for mixed tensor") {
@@ -71,10 +71,10 @@ TEST("require that tensor peek is not optimized for mixed tensor") {
TEST("require that indexes are truncated when converted to integers") {
TEST_DO(verify("x3{x:(a+0.7)}", 2.0, 1, 0));
TEST_DO(verify("x3{x:(a+0.3)}", 2.0, 1, 0));
- TEST_DO(verify("xm{x:(a+0.7)}", 1.0, 0, 1));
- TEST_DO(verify("xm{x:(a+0.3)}", 1.0, 0, 1));
- TEST_DO(verify("xm{x:(-a-0.7)}", 4.0, 0, 1));
- TEST_DO(verify("xm{x:(-a-0.3)}", 4.0, 0, 1));
+ TEST_DO(verify("xm{x:(a+0.7)}", 1.0, 0, 0));
+ TEST_DO(verify("xm{x:(a+0.3)}", 1.0, 0, 0));
+ TEST_DO(verify("xm{x:(-a-0.7)}", 4.0, 0, 0));
+ TEST_DO(verify("xm{x:(-a-0.3)}", 4.0, 0, 0));
}
TEST_MAIN() { TEST_RUN_ALL(); }