From e3c287570f1a89b94e2f3ae10389a3327b1dfd32 Mon Sep 17 00:00:00 2001
From: Wei Ziran <weiziran125@gmail.com>
Date: Mon, 10 Feb 2025 11:08:37 +0800
Subject: [PATCH] fix oom

---
 pkg/vm/engine/tae/mergesort/merger.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pkg/vm/engine/tae/mergesort/merger.go b/pkg/vm/engine/tae/mergesort/merger.go
index 2b96b0486ec43..8f4a78031bee4 100644
--- a/pkg/vm/engine/tae/mergesort/merger.go
+++ b/pkg/vm/engine/tae/mergesort/merger.go
@@ -142,6 +142,8 @@ func newMerger[T comparable](host MergeTaskHost, lessFunc sort.LessFunc[T], sort
 }
 
 func (m *merger[T]) merge(ctx context.Context) error {
+	defer m.release()
+
 	for i := 0; i < m.objCnt; i++ {
 		if ok, err := m.loadBlk(ctx, uint32(i)); !ok {
 			if err == nil {
@@ -156,7 +158,6 @@ func (m *merger[T]) merge(ctx context.Context) error {
 			src:    uint32(i),
 		})
 	}
-	defer m.release()
 
 	var releaseF func()
 	for i := 0; i < m.objCnt; i++ {