Commit b5ab8ca1 authored by nmostafa's avatar nmostafa

style-apply

parent c6e9747b
...@@ -61,7 +61,9 @@ namespace ngraph ...@@ -61,7 +61,9 @@ namespace ngraph
using TypeList = llvm::SmallVector<mlir::Type, 4>; using TypeList = llvm::SmallVector<mlir::Type, 4>;
MLIRCompiler(const ngraph::op::CompiledKernel* compiled_kernel) MLIRCompiler(const ngraph::op::CompiledKernel* compiled_kernel)
: m_compiled_kernel(compiled_kernel) {} : m_compiled_kernel(compiled_kernel)
{
}
/// Set runtime tensor arguments for the sub-graph /// Set runtime tensor arguments for the sub-graph
void set_args(std::vector<void*>* external_tensors); void set_args(std::vector<void*>* external_tensors);
......
...@@ -66,7 +66,8 @@ namespace ngraph ...@@ -66,7 +66,8 @@ namespace ngraph
ptr_args.push_back(ctx->buffer_data[buffer_index]); ptr_args.push_back(ctx->buffer_data[buffer_index]);
} }
// Compile nodes within the CompiledKernel op. // Compile nodes within the CompiledKernel op.
CompiledKernel* compiled_kernel = static_cast<CompiledKernel*>(const_cast<Node*>(node)); CompiledKernel* compiled_kernel =
static_cast<CompiledKernel*>(const_cast<Node*>(node));
bool is_module_ready = true; bool is_module_ready = true;
auto it = ctx->mlir_compilers.find(compiled_kernel); auto it = ctx->mlir_compilers.find(compiled_kernel);
......
...@@ -25,8 +25,8 @@ ...@@ -25,8 +25,8 @@
#include <tbb/flow_graph.h> #include <tbb/flow_graph.h>
#include <tbb/global_control.h> #include <tbb/global_control.h>
#include <tbb/task_scheduler_init.h> #include <tbb/task_scheduler_init.h>
#include "ngraph/op/experimental/compiled_kernel.hpp"
#include "contrib/mlir/compiler.hpp" #include "contrib/mlir/compiler.hpp"
#include "ngraph/op/experimental/compiled_kernel.hpp"
namespace mkldnn namespace mkldnn
{ {
...@@ -69,7 +69,9 @@ namespace ngraph ...@@ -69,7 +69,9 @@ namespace ngraph
std::set<size_t> breakpoints; std::set<size_t> breakpoints;
size_t pc; size_t pc;
#ifdef NGRAPH_MLIR_ENABLE #ifdef NGRAPH_MLIR_ENABLE
std::unordered_map<ngraph::op::CompiledKernel*, ngraph::runtime::ngmlir::MLIRCompiler> mlir_compilers; std::unordered_map<ngraph::op::CompiledKernel*,
ngraph::runtime::ngmlir::MLIRCompiler>
mlir_compilers;
#endif #endif
}; };
} }
......
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