Skip to content

GitLab

Commit 733591e1 authored by Paul's avatar Paul
Browse files

Formatting

parent 117ae108
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 6 deletions
src/eliminate_allocation.cpp
View file @ 733591e1
...@@ -17,7 +17,7 @@ void eliminate_allocation::apply(program& p) const ...@@ -17,7 +17,7 @@ void eliminate_allocation::apply(program& p) const
if(ins->op.name() != allocation_op) if(ins->op.name() != allocation_op)
continue; continue;
allocs.emplace_back(ins, n); allocs.emplace_back(ins, n);
std::size_t size = ins->get_shape().bytes(); std::size_t size = ins->get_shape().bytes();
std::size_t padding = (alignment - (size % alignment)) % alignment; std::size_t padding = (alignment - (size % alignment)) % alignment;
n += size + padding; n += size + padding;
} }
......
test/eliminate_allocation_test.cpp
View file @ 733591e1
...@@ -24,7 +24,9 @@ struct allocate ...@@ -24,7 +24,9 @@ struct allocate
migraph::check_shapes{inputs}.has(0); migraph::check_shapes{inputs}.has(0);
return s; return s;
} }
migraph::argument compute(migraph::context&, const migraph::shape& output_shape, const std::vector<migraph::argument>&) const migraph::argument compute(migraph::context&,
const migraph::shape& output_shape,
const std::vector<migraph::argument>&) const
{ {
return {output_shape}; return {output_shape};
} }
...@@ -44,7 +46,7 @@ void basic() ...@@ -44,7 +46,7 @@ void basic()
p.compile(eliminate_allocation_target{}); p.compile(eliminate_allocation_target{});
EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}}); EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}});
EXPECT(p.get_parameter_shape("memory").bytes() == (8*4 + 40*4 + 200*4)); EXPECT(p.get_parameter_shape("memory").bytes() == (8 * 4 + 40 * 4 + 200 * 4));
} }
void aligned() void aligned()
...@@ -61,7 +63,7 @@ void aligned() ...@@ -61,7 +63,7 @@ void aligned()
p.compile(eliminate_allocation_target{}); p.compile(eliminate_allocation_target{});
EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}}); EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}});
EXPECT(p.get_parameter_shape("memory").bytes() == (32 + 32 + 200*4)); EXPECT(p.get_parameter_shape("memory").bytes() == (32 + 32 + 200 * 4));
} }
void unaligned() void unaligned()
...@@ -78,7 +80,7 @@ void unaligned() ...@@ -78,7 +80,7 @@ void unaligned()
p.compile(eliminate_allocation_target{1}); p.compile(eliminate_allocation_target{1});
EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}}); EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}});
EXPECT(p.get_parameter_shape("memory").bytes() == (1*4 + 2*4 + 200*4)); EXPECT(p.get_parameter_shape("memory").bytes() == (1 * 4 + 2 * 4 + 200 * 4));
} }
void float_aligned() void float_aligned()
...@@ -95,7 +97,7 @@ void float_aligned() ...@@ -95,7 +97,7 @@ void float_aligned()
p.compile(eliminate_allocation_target{4}); p.compile(eliminate_allocation_target{4});
EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}}); EXPECT(p.get_shape() == migraph::shape{migraph::shape::float_type, {200}});
EXPECT(p.get_parameter_shape("memory").bytes() == (1*4 + 2*4 + 200*4)); EXPECT(p.get_parameter_shape("memory").bytes() == (1 * 4 + 2 * 4 + 200 * 4));
} }
int main() int main()
......
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