Minor typo fix: in not implemented message for the exclusive and reve…

…rse attributes for cumsum (#2740)
llvm · Jan 10, 2024 · aee1fca · aee1fca
1 parent 2956971
commit aee1fca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp b/lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp
@@ -952,11 +952,11 @@ void mlir::torch::onnx_c::populateDefaultDomainAtoF(
         int64_t exclusive;
         int64_t reverse;
         // if bind succeeds and either is set, fail because not implemented
-        if (binder.s64IntegerAttr(exclusive, "exclusive", 0))
+        if (!binder.s64IntegerAttr(exclusive, "exclusive", 0))
           if (exclusive != 0)
             return rewriter.notifyMatchFailure(
                 binder.op, "unsupported onnx.CumSum conversion: exclusive");
-        if (binder.s64IntegerAttr(reverse, "reverse", 0))
+        if (!binder.s64IntegerAttr(reverse, "reverse", 0))
           if (reverse != 0)
             return rewriter.notifyMatchFailure(
                 binder.op, "unsupported onnx.CumSum conversion: reverse");