• Diego Caballero's avatar
    [MLIR] Improve gather op lowering (#3667) · 3aa4db1d
    Diego Caballero authored
    * [MLIR] Add support for parsing nGraph tensor type
    
    Initial commit that enables nGraph parsing. It's needed for testing.
    
    * [MLIR] Enable nGraph dialect in ngraph-opt
    
    This PR registers nGraph dialect in ngraph-opt and prepares
    nGraph lowering pass for LIT testing, fixing all the related issues.
    Among other things, lowering pass has to be turned into a function pass,
    dead argument in constructor was removed and `convert-ngraph-to-affine`
    flag was added.
    
    * Fix issue with function name and multiple functions
    
    * Extend module_function.mlir lit test
    
    * [MLIR] Add support for parsing nGraph element types
    
    It introduces initial support for parsing nGraph signed/unsigned
    integer and floating point data types.
    
    * [MLIR] Improve gather op lowering
    
    This PR interchanges indices and param loops in gather lowering so
    that a better memory access patter is generated. Fusion of gather
    with other ops is also observed with this change.
    3aa4db1d
Name
Last commit
Last update
..
backend Loading commit data...
files Loading commit data...
mlir Loading commit data...
models Loading commit data...
onnx Loading commit data...
opset_pass Loading commit data...
ref_generators Loading commit data...
type_prop Loading commit data...
util Loading commit data...
CMakeLists.txt Loading commit data...
algebraic_simplification.cpp Loading commit data...
aligned_buffer.cpp Loading commit data...
all_close_f.cpp Loading commit data...
assertion.cpp Loading commit data...
backend_api.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_performance.cpp Loading commit data...
bfloat16.cpp Loading commit data...
build_graph.cpp Loading commit data...
builder.cpp Loading commit data...
builder_autobroadcast.cpp Loading commit data...
builder_quantization.cpp Loading commit data...
check.cpp Loading commit data...
concat_fusion.cpp Loading commit data...
constant_folding.cpp Loading commit data...
control_dependencies.cpp Loading commit data...
coordinate.cpp Loading commit data...
copy.cpp Loading commit data...
core.cpp Loading commit data...
core_fusion.cpp Loading commit data...
cpio.cpp Loading commit data...
cpu_codegen.cpp Loading commit data...
cpu_debug_tracer.cpp Loading commit data...
cpu_debugger.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_test.cpp Loading commit data...
cse.cpp Loading commit data...
dyn_elimination.cpp Loading commit data...
element_type.cpp Loading commit data...
event_tracing.cpp Loading commit data...
file_util.cpp Loading commit data...
float16.cpp Loading commit data...
halide.cpp Loading commit data...
includes.cpp Loading commit data...
input_output_assign.cpp Loading commit data...
main.cpp Loading commit data...
misc.cpp Loading commit data...
misc.hpp Loading commit data...
mkldnn.cpp Loading commit data...
ngraph_api.cpp Loading commit data...
node_input_output.cpp Loading commit data...
nop_elimination.cpp Loading commit data...
op.cpp Loading commit data...
partial_shape.cpp Loading commit data...
pass.cpp Loading commit data...
pass_liveness.cpp Loading commit data...
pass_manager.cpp Loading commit data...
pass_memory_layout.cpp Loading commit data...
pass_shape_relevance.cpp Loading commit data...
pattern.cpp Loading commit data...
provenance.cpp Loading commit data...
replace_node.cpp Loading commit data...
reshape_elimination.cpp Loading commit data...
reshape_sinking.cpp Loading commit data...
serialize.cpp Loading commit data...
shape.cpp Loading commit data...
specialize_function.cpp Loading commit data...
tensor.cpp Loading commit data...
tools.cpp Loading commit data...
type_prop_benchmark.cpp Loading commit data...
type_prop_layers.cpp Loading commit data...
update_convolution_reference.sh Loading commit data...
update_dyn_replace_slice_reference.sh Loading commit data...
update_dyn_slice_reference.sh Loading commit data...
util.cpp Loading commit data...
zero_dim_tensor_elimination.cpp Loading commit data...