Skip to content

GitLab

Commit 11d00d61 authored by Paul's avatar Paul
Browse files

Formatting

parent 3d1ae7d1
Show whitespace changes
Inline Side-by-side
Showing with 14 additions and 11 deletions
test/gpu/literal.cpp
View file @ 11d00d61
...@@ -13,10 +13,13 @@ void gpu_literal_test() ...@@ -13,10 +13,13 @@ void gpu_literal_test()
p.add_literal(lit); p.add_literal(lit);
p.compile(migraph::gpu::target{}); p.compile(migraph::gpu::target{});
auto scratch = p.get_parameter("scratch"); auto scratch = p.get_parameter("scratch");
if(scratch == p.end()) { if(scratch == p.end())
{
auto result = p.eval({}); auto result = p.eval({});
EXPECT(lit == migraph::gpu::from_gpu(result)); EXPECT(lit == migraph::gpu::from_gpu(result));
} else { }
else
{
EXPECT(scratch->get_shape().bytes() == lit.get_shape().bytes()); EXPECT(scratch->get_shape().bytes() == lit.get_shape().bytes());
} }
} }
......
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