Skip to content

N
ngraph
Unverified Commit 29d29162 authored by Adam Procter's avatar Adam Procter Committed by GitHub
Browse files
Options

Merge pull request #2318 from NervanaSystems/mkarzyns/disable_cpu_test 

[ONNX] Move CPU tests to INTERPRETER
parents d8b2dd79 a9291030
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/onnx_import.cpp
View file @ 29d29162
...@@ -1660,7 +1660,7 @@ TEST(onnx, model_argmax_int32) ...@@ -1660,7 +1660,7 @@ TEST(onnx, model_argmax_int32)
std::vector<std::int64_t>{1, 1, 1, 1, 1, 1}}; std::vector<std::int64_t>{1, 1, 1, 1, 1, 1}};
std::vector<std::vector<std::int64_t>> outputs{ std::vector<std::vector<std::int64_t>> outputs{
execute<std::int32_t, std::int64_t>(function, inputs, "CPU")}; execute<std::int32_t, std::int64_t>(function, inputs, "INTERPRETER")};
EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front())); EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front()));
} }
...@@ -1675,6 +1675,6 @@ TEST(onnx, model_argmin_int32) ...@@ -1675,6 +1675,6 @@ TEST(onnx, model_argmin_int32)
std::vector<std::vector<std::int64_t>> expected_output{std::vector<std::int64_t>{0, 0, 0, 0}}; std::vector<std::vector<std::int64_t>> expected_output{std::vector<std::int64_t>{0, 0, 0, 0}};
std::vector<std::vector<std::int64_t>> outputs{ std::vector<std::vector<std::int64_t>> outputs{
execute<std::int32_t, std::int64_t>(function, inputs, "CPU")}; execute<std::int32_t, std::int64_t>(function, inputs, "INTERPRETER")};
EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front())); EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front()));
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment