isl_test.c: before_for: return early on error

[platform/upstream/isl.git] / isl_ast_codegen.c

diff --git a/isl_ast_codegen.c b/isl_ast_codegen.c
index e0e217d..c2eebe7 100644 (file)
--- a/isl_ast_codegen.c
+++ b/isl_ast_codegen.c
@@ -291,7 +291,7 @@ static __isl_give isl_ast_graft *after_each_for(__isl_keep isl_ast_graft *graft,
        __isl_keep isl_ast_build *build)
 {
        if (!graft || !build)
-               isl_ast_graft_free(graft);
+               return isl_ast_graft_free(graft);
        if (!build->after_each_for)
                return graft;
        graft->node = build->after_each_for(graft->node, build,