Commit 6c676d2d authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

CPU: Merge fixes

parent 2659d5be
...@@ -226,20 +226,13 @@ void runtime::cpu::CPU_ExternalFunction::compile() ...@@ -226,20 +226,13 @@ void runtime::cpu::CPU_ExternalFunction::compile()
string function_name = m_function->get_name(); string function_name = m_function->get_name();
<<<<<<< HEAD
ngraph::pass::Manager pass_manager; ngraph::pass::Manager pass_manager;
pass_manager.register_pass<runtime::cpu::pass::CPUFusion>();
pass_manager.register_pass<runtime::cpu::pass::CPULayout>(); pass_manager.register_pass<runtime::cpu::pass::CPULayout>();
pass_manager.register_pass<ngraph::pass::Liveness>(); pass_manager.register_pass<ngraph::pass::Liveness>();
pass_manager.register_pass<ngraph::pass::MemoryLayout>(MemoryPoolAlignment); pass_manager.register_pass<ngraph::pass::MemoryLayout>(MemoryPoolAlignment);
=======
pass::Manager pass_manager;
// For now, just make everyone row-major.
pass_manager.register_pass<pass::CPUFusion>();
pass_manager.register_pass<pass::AssignLayout<descriptor::layout::DenseTensorViewLayout>>();
pass_manager.register_pass<pass::Liveness>();
pass_manager.register_pass<pass::MemoryLayout>(64);
>>>>>>> master
pass_manager.run_passes(m_function); pass_manager.run_passes(m_function);
codegen::CodeWriter writer; codegen::CodeWriter writer;
......
This diff is collapsed.
// ---------------------------------------------------------------------------- // ----------------------------------------------------------------------------
// Copyright 2018 Nervana Systems Inc. // Copyright 2018 Nervana Systems Inc.
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
// You may obtain a copy of the License at // You may obtain a copy of the License at
// //
// http://www.apache.org/licenses/LICENSE-2.0 // http://www.apache.org/licenses/LICENSE-2.0
// //
// Unless required by applicable law or agreed to in writing, software // Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, // distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// ---------------------------------------------------------------------------- // ----------------------------------------------------------------------------
#pragma once #pragma once
#include "ngraph/pass/graph_rewrite.hpp" #include "ngraph/pass/graph_rewrite.hpp"
namespace ngraph namespace ngraph
{ {
namespace pass namespace runtime
{ {
class CPUFusion; namespace cpu
} {
} namespace pass
{
class ngraph::pass::CPUFusion : public ngraph::pass::GraphRewrite class CPUFusion;
{ }
public: }
CPUFusion() }
: GraphRewrite() }
{
construct_gemm_pattern(); class ngraph::runtime::cpu::pass::CPUFusion : public ngraph::pass::GraphRewrite
} {
public:
private: CPUFusion()
void construct_gemm_pattern(); : GraphRewrite()
}; {
construct_gemm_pattern();
}
private:
void construct_gemm_pattern();
};
This diff is collapsed.
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