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
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()
{
// Lower NG dialect to Affine
......@@ -284,7 +284,7 @@ void MLIRCompiler::lower_ng_dialect()
dump_mlir_module("LLVM-IR Dialect Dump:");
// Lower to LLVM BC and optimize
// Lower to LLVM BC and optimize
// Initialize LLVM targets.
llvm::InitializeNativeTarget();
llvm::InitializeNativeTargetAsmPrinter();
......
......@@ -61,7 +61,9 @@ namespace ngraph
using TypeList = llvm::SmallVector<mlir::Type, 4>;
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
void set_args(std::vector<void*>* external_tensors);
......
......@@ -66,7 +66,8 @@ namespace ngraph
ptr_args.push_back(ctx->buffer_data[buffer_index]);
}
// 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;
auto it = ctx->mlir_compilers.find(compiled_kernel);
......
......@@ -25,8 +25,8 @@
#include <tbb/flow_graph.h>
#include <tbb/global_control.h>
#include <tbb/task_scheduler_init.h>
#include "ngraph/op/experimental/compiled_kernel.hpp"
#include "contrib/mlir/compiler.hpp"
#include "ngraph/op/experimental/compiled_kernel.hpp"
namespace mkldnn
{
......@@ -69,7 +69,9 @@ namespace ngraph
std::set<size_t> breakpoints;
size_t pc;
#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
};
}
......
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