Skip to content

GitLab

Commit 4a880d96 authored by Paul's avatar Paul
Browse files

Update last test

parent d8597de7
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
test/memory_coloring_test.cpp
View file @ 4a880d96
...@@ -298,14 +298,14 @@ void test19() ...@@ -298,14 +298,14 @@ void test19()
void test20() void test20()
{ {
migraph::program p; migraph::program p;
auto a1 = add_alloc(p, {migraph::shape::float_type, {8}}); auto a1 = add_alloc(p, {migraph::shape::float_type, {32}});
auto p1 = p.add_instruction(pass_op{}, a1); auto a2 = add_alloc(p, {migraph::shape::float_type, {32}});
auto a2 = add_alloc(p, {migraph::shape::float_type, {40}}); auto a3 = add_alloc(p, {migraph::shape::float_type, {32}});
auto p2 = p.add_instruction(pass_op{}, p1, a2); auto p1 = p.add_instruction(pass_op{}, a1, a2, a3);
auto a3 = add_alloc(p, {migraph::shape::float_type, {40}}); auto a4 = add_alloc(p, {migraph::shape::float_type, {32}});
p.add_instruction(pass_op{}, a3, p2, p1); p.add_instruction(pass_op{}, a4, p1);
p.compile(memory_coloring_target{}); p.compile(memory_coloring_target{});
EXPECT(p.get_parameter_shape("scratch").bytes() == 192); EXPECT(p.get_parameter_shape("scratch").bytes() == 384);
EXPECT(no_allocate(p)); EXPECT(no_allocate(p));
} }
......
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