"cacheflow/vscode:/vscode.git/clone" did not exist on "46ce1356f7108f27af9ebb153ddbe02d1a3fa97d"
Unverified Commit a761ffaa authored by Krzysztof Drewniak's avatar Krzysztof Drewniak Committed by GitHub
Browse files

[mlir] Apply is_mlir_conv predicate in standalone MLIr offloading (#2249)

Currently, the is_mlir_conv predicate wasn't being used when
offloading standalone convolutions to MLIR on Navi3x, which caused
failures relating to being unable to construct the MLIR program when a
3D convlolution was passed in.

This commit amends the standalone lowering to use said predicate, as
well as to include quant_convolution and quant_dot into the set of
operations that get a standalone lowering.
parent 75a73214
...@@ -327,12 +327,12 @@ struct find_mlir_standalone_op ...@@ -327,12 +327,12 @@ struct find_mlir_standalone_op
struct find_mlir_standalone_convolution_op : find_mlir_standalone_op struct find_mlir_standalone_convolution_op : find_mlir_standalone_op
{ {
auto matcher() const { return match::name("convolution"); } auto matcher() const { return is_mlir_conv; }
}; };
struct find_mlir_standalone_dot_op : find_mlir_standalone_op struct find_mlir_standalone_dot_op : find_mlir_standalone_op
{ {
auto matcher() const { return match::name("dot"); } auto matcher() const { return match::any_of(match::name("dot"), match::name("quant_dot")); }
}; };
/** /**
...@@ -365,7 +365,7 @@ bool is_enabled(std::string_view op_name, context* ctx) ...@@ -365,7 +365,7 @@ bool is_enabled(std::string_view op_name, context* ctx)
{ {
return true; return true;
} }
else if(op_name == "convolution") else if(op_name == "convolution" or op_name == "quant_convolution")
{ {
if(ctx == nullptr) if(ctx == nullptr)
{ {
......
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