"src/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "4614de7c95f076304c8cbc65588e67cf0e9a88db"
Commit 14d40d9b authored by charlie's avatar charlie
Browse files

Update ref_ops_test

parent 8bbec307
...@@ -925,32 +925,6 @@ TEST_CASE(contiguous_test) ...@@ -925,32 +925,6 @@ TEST_CASE(contiguous_test)
EXPECT(migraphx::verify_range(results_vector, data)); EXPECT(migraphx::verify_range(results_vector, data));
} }
TEST_CASE(contiguous_param_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
migraphx::shape a_shape{migraphx::shape::float_type, {1, 3, 2, 2}, {12, 1, 6, 3}};
auto a = mm->add_parameter("X", a_shape);
mm->add_instruction(migraphx::make_op("contiguous"), a);
p.compile(migraphx::ref::target{});
std::vector<float> data(12);
std::iota(data.begin(), data.end(), 0);
migraphx::parameter_map params;
params["X"] = migraphx::argument(a_shape, data.data());
auto result = p.eval(params).back();
result.visit([&](auto output) {
std::vector<size_t> new_strides = {12, 4, 2, 1};
EXPECT(bool{output.get_shape().strides() == new_strides});
});
std::vector<float> results_vector(12);
result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
std::vector<float> gold = {0, 3, 6, 9, 1, 4, 7, 10, 2, 5, 8, 11};
EXPECT(migraphx::verify_range(results_vector, gold));
}
TEST_CASE(contiguous_dyn_test) TEST_CASE(contiguous_dyn_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -968,10 +942,8 @@ TEST_CASE(contiguous_dyn_test) ...@@ -968,10 +942,8 @@ TEST_CASE(contiguous_dyn_test)
params["X"] = migraphx::argument(static_shape, data.data()); params["X"] = migraphx::argument(static_shape, data.data());
auto result = p.eval(params).back(); auto result = p.eval(params).back();
result.visit([&](auto output) { std::vector<size_t> new_strides = {12, 4, 2, 1};
std::vector<size_t> new_strides = {12, 4, 2, 1}; EXPECT(result.get_shape().strides() == new_strides);
EXPECT(bool{output.get_shape().strides() == new_strides});
});
std::vector<float> results_vector(12); std::vector<float> results_vector(12);
result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); }); result.visit([&](auto output) { results_vector.assign(output.begin(), output.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