[Mlir-commits] [mlir] 36a6e56 - Fix possible memory leak in a MLIR unit-test

[Mehdi Amini]

Author: Mehdi Amini
Date: 2022-01-01T01:42:26Z
New Revision: 36a6e56bff7759f14160ad7413b69cb1eadc2fc2

URL: https://github.com/llvm/llvm-project/commit/36a6e56bff7759f14160ad7413b69cb1eadc2fc2
DIFF: https://github.com/llvm/llvm-project/commit/36a6e56bff7759f14160ad7413b69cb1eadc2fc2.diff

LOG: Fix possible memory leak in a MLIR unit-test

Flagged by Coverity

Added: 
    

Modified: 
    mlir/test/CAPI/ir.c

Removed: 
    


################################################################################
diff  --git a/mlir/test/CAPI/ir.c b/mlir/test/CAPI/ir.c
index ac74c5748286d..51e0c2b4bcd24 100644
--- a/mlir/test/CAPI/ir.c
+++ b/mlir/test/CAPI/ir.c
@@ -276,8 +276,10 @@ int collectStats(MlirOperation operation) {
 
   do {
     int retval = collectStatsSingle(head, &stats);
-    if (retval)
+    if (retval) {
+      free(head);
       return retval;
+    }
     OpListNode *next = head->next;
     free(head);
     head = next;