Commit 64e975c6 authored by Diego Caballero's avatar Diego Caballero Committed by Scott Cyphers

[MLIR] Disable python rounding Divide op in MLIR (#3440)

Python rounding is not supported yet in nGraph dialect.
parent f70e2174
......@@ -279,6 +279,18 @@ bool MLIRSubgraphExtractionPass::is_supported_mlir_op(std::shared_ptr<Node> node
// check on invariants expected by MLIR backend
if (TI(ngraph::op::Divide) == TI(*node))
{
auto* div = static_cast<ngraph::op::Divide*>(node.get());
if (div->is_pythondiv())
{
// Python specific division rounding is not supported yet.
return false;
}
return true;
}
// Dot is 2D only
if (TI(ngraph::op::Dot) == TI(*node))
{
......
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