"docs/source/tutorials/pruning_quick_start_mnist.py" did not exist on "cef9babdb74b6f157fc8d684f0c352c3d1b3a49f"
Commit 29104134 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Change up tests to huge of static and random data

Remove the need to use gpu, switch this to ref.
change names to reflect static vs random data
parent d8eb00b4
Pipeline #669 failed with stages
in 0 seconds
...@@ -5138,7 +5138,7 @@ TEST_CASE(nms_test) ...@@ -5138,7 +5138,7 @@ TEST_CASE(nms_test)
EXPECT(migraphx::verify_range(result, gold)); EXPECT(migraphx::verify_range(result, gold));
} }
TEST_CASE(nms_huge_test_cpu) TEST_CASE(nms_huge_test_random_data)
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
...@@ -5193,7 +5193,7 @@ TEST_CASE(nms_huge_test_cpu) ...@@ -5193,7 +5193,7 @@ TEST_CASE(nms_huge_test_cpu)
EXPECT(true); EXPECT(true);
} }
TEST_CASE(nms_huge_test_gpu) TEST_CASE(nms_huge_test_static_data)
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
...@@ -5231,7 +5231,7 @@ TEST_CASE(nms_huge_test_gpu) ...@@ -5231,7 +5231,7 @@ TEST_CASE(nms_huge_test_gpu)
); );
mm->add_return({r}); mm->add_return({r});
p.compile(migraphx::make_target("gpu")); p.compile(migraphx::make_target("ref"));
auto output = p.eval({}).back(); auto output = p.eval({}).back();
// std::vector<int64_t> result; // std::vector<int64_t> result;
// output.visit([&](auto out) { result.assign(out.begin(), out.end()); }); // output.visit([&](auto out) { result.assign(out.begin(), out.end()); });
......
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