Unverified Commit 88b1ff33 authored by Jayaram Bobba's avatar Jayaram Bobba Committed by GitHub

Merge branch 'master' into jmenon/dex4

parents b9425e49 e18aa2fa
...@@ -435,6 +435,7 @@ void codegen::CompilerCore::configure_search_path() ...@@ -435,6 +435,7 @@ void codegen::CompilerCore::configure_search_path()
add_header_search_path("/usr/lib/gcc/x86_64-linux-gnu/" + header_version + "/include-fixed"); add_header_search_path("/usr/lib/gcc/x86_64-linux-gnu/" + header_version + "/include-fixed");
add_header_search_path("/usr/include/x86_64-linux-gnu"); add_header_search_path("/usr/include/x86_64-linux-gnu");
add_header_search_path("/usr/include"); add_header_search_path("/usr/include");
add_header_search_path("/usr/include/c++/" + header_version + "/x86_64-redhat-linux");
add_header_search_path(EIGEN_HEADERS_PATH); add_header_search_path(EIGEN_HEADERS_PATH);
add_header_search_path(MKLDNN_HEADERS_PATH); add_header_search_path(MKLDNN_HEADERS_PATH);
......
...@@ -1652,9 +1652,7 @@ namespace ngraph ...@@ -1652,9 +1652,7 @@ namespace ngraph
void CPU_Emitter::EMITTER_DECL(ngraph::op::Reshape) void CPU_Emitter::EMITTER_DECL(ngraph::op::Reshape)
{ {
auto reshape = static_cast<const ngraph::op::Reshape*>(node); auto reshape = static_cast<const ngraph::op::Reshape*>(node);
auto annotation = reshape->get_op_annotations(); if (!reshape->get_is_transpose() && out[0].get_name() == args[0].get_name())
if (annotation && annotation->get_in_place_oi_pairs().size() > 0 &&
out[0].get_name() == args[0].get_name())
{ {
writer.block_begin(); writer.block_begin();
writer << "// Stride change only, skipping.\n"; writer << "// Stride change only, skipping.\n";
......
...@@ -548,13 +548,7 @@ namespace ngraph ...@@ -548,13 +548,7 @@ namespace ngraph
// not in the memory pool, or has output users. // not in the memory pool, or has output users.
bool need_copy = bool need_copy =
reshape->get_is_transpose() || arg->is_parameter() || arg->is_constant(); reshape->get_is_transpose() || arg->is_parameter() || arg->is_constant();
for (auto n = users.begin(); !need_copy && n != users.end(); ++n)
{
if ((*n)->is_output())
{
need_copy = true;
}
}
if (!need_copy) if (!need_copy)
{ {
// map output to the input memory // map output to the input memory
......
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