Commit b5ab8ca1 authored by nmostafa's avatar nmostafa

style-apply

parent c6e9747b
...@@ -239,7 +239,7 @@ MLIRCompiler::TensorInfo MLIRCompiler::get_tensor_value(descriptor::Tensor* tens ...@@ -239,7 +239,7 @@ MLIRCompiler::TensorInfo MLIRCompiler::get_tensor_value(descriptor::Tensor* tens
return it->second; return it->second;
} }
// Lowers nGraph dialect all the way to LLVM module. // Lowers nGraph dialect all the way to LLVM module.
void MLIRCompiler::lower_ng_dialect() void MLIRCompiler::lower_ng_dialect()
{ {
// Lower NG dialect to Affine // Lower NG dialect to Affine
...@@ -284,7 +284,7 @@ void MLIRCompiler::lower_ng_dialect() ...@@ -284,7 +284,7 @@ void MLIRCompiler::lower_ng_dialect()
dump_mlir_module("LLVM-IR Dialect Dump:"); dump_mlir_module("LLVM-IR Dialect Dump:");
// Lower to LLVM BC and optimize // Lower to LLVM BC and optimize
// Initialize LLVM targets. // Initialize LLVM targets.
llvm::InitializeNativeTarget(); llvm::InitializeNativeTarget();
llvm::InitializeNativeTargetAsmPrinter(); llvm::InitializeNativeTargetAsmPrinter();
......
...@@ -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