Commit 026e48a3 authored by mei-ye's avatar mei-ye
Browse files

remove write_literal and add env to disable mem color for eliminate_allocation_test.cpp

parent 5a335544
...@@ -550,18 +550,6 @@ struct load ...@@ -550,18 +550,6 @@ struct load
} }
}; };
struct write_literal
{
std::size_t offset = 0;
bool pre_copy = false;
std::string name() const { return "write_literal"; }
shape compute_shape(std::vector<shape> inputs) const { return inputs.at(1); }
argument compute(context&, const shape&, const std::vector<argument>&) const
{
MIGRAPH_THROW("not computable");
}
};
struct outline struct outline
{ {
shape s; shape s;
......
...@@ -234,10 +234,12 @@ void memory_coloring_impl::rewrite() ...@@ -234,10 +234,12 @@ void memory_coloring_impl::rewrite()
} }
else if(is_literal(ins)) else if(is_literal(ins))
{ {
#if 0
auto pre = p_program->add_literal(ins->lit); auto pre = p_program->add_literal(ins->lit);
bool pre_copy = (interval->get_begin() < earliest_end_point); bool pre_copy = (interval->get_begin() < earliest_end_point);
p_program->replace_instruction( p_program->replace_instruction(
ins, write_literal{offset, pre_copy}, scratch_param, pre); ins, write_literal{offset, pre_copy}, scratch_param, pre);
#endif
} }
} }
} }
......
...@@ -102,6 +102,7 @@ void float_aligned() ...@@ -102,6 +102,7 @@ void float_aligned()
int main() int main()
{ {
setenv("MIGRAPH_DISABLE_MEMORY_COLORING", "1", 1);
basic(); basic();
aligned(); aligned();
unaligned(); unaligned();
......
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