Commit c8712215 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent ad774473
......@@ -16,17 +16,15 @@ struct lowering_target
std::string name() const { return "gpu::lowering"; }
std::vector<migraphx::pass> get_passes(migraphx::context& gctx) const
{
auto &ctx = migraphx::any_cast<migraphx::gpu::context>(gctx);
auto& ctx = migraphx::any_cast<migraphx::gpu::context>(gctx);
return {migraphx::gpu::lowering{ctx}, migraphx::dead_code_elimination{}};
}
migraphx::gpu::context get_context() const { return migraphx::gpu::context{}; }
};
TEST_CASE(trans_tanh)
{
auto create_program = []{
auto create_program = [] {
migraphx::program p;
migraphx::shape s{migraphx::shape::float_type, {2, 3}};
auto x = p.add_parameter("x", s);
......@@ -42,15 +40,15 @@ TEST_CASE(trans_tanh)
// relace the add instruction with using a incorrect
// output shape
for (auto ins : iterator_for(p1))
for(auto ins : iterator_for(p1))
{
if (ins->name() == "add")
if(ins->name() == "add")
{
migraphx::shape wrong_s{migraphx::shape::float_type, {3, 2}};
migraphx::instruction::replace(ins, ins->get_operator(), wrong_s, ins->inputs());
}
if (ins->name() == "tanh")
if(ins->name() == "tanh")
{
migraphx::shape orig_s{migraphx::shape::float_type, {2, 3}};
migraphx::instruction::replace(ins, ins->get_operator(), orig_s, ins->inputs());
......@@ -62,9 +60,9 @@ TEST_CASE(trans_tanh)
p2.compile(lowering_target{});
EXPECT(p1 != p2);
for (auto ins : iterator_for(p1))
for(auto ins : iterator_for(p1))
{
if (ins->name() == "gpu::add")
if(ins->name() == "gpu::add")
{
migraphx::shape correct_s{migraphx::shape::float_type, {2, 3}};
migraphx::instruction::replace(ins, ins->get_operator(), correct_s, ins->inputs());
......@@ -72,7 +70,8 @@ TEST_CASE(trans_tanh)
}
EXPECT(p1 != p2);
migraphx::run_passes(p1, {migraphx::gpu::adjust_allocation{}, migraphx::dead_code_elimination{}});
migraphx::run_passes(p1,
{migraphx::gpu::adjust_allocation{}, migraphx::dead_code_elimination{}});
EXPECT(p1 == p2);
}
......
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