Fix freeing of args

180329bf · nmostafa · 048bff67 · 180329bf
Commit 180329bf authored Oct 10, 2019 by nmostafa
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

compiler.cpp src/contrib/mlir/compiler/compiler.cpp +2 -0

No files found.
--- a/src/contrib/mlir/compiler/compiler.cpp
+++ b/src/contrib/mlir/compiler/compiler.cpp
@@ -777,6 +777,8 @@ void MLIRCompiler::cleanup()
    // Free void double pointer arguments without freeing external tensor data.
    for (auto* arg : m_invokeArgs)
    {
+        auto* memRefArg = *((mlir::StaticFloatMemRef**)arg);
+        free(memRefArg);
        free(arg);
    }