From 51b9f0b82ab11a79fafb9b499017214f59e32bf3 Mon Sep 17 00:00:00 2001 From: Mehdi Amini Date: Sun, 3 Oct 2021 03:27:54 +0000 Subject: [PATCH] Fix memory leaks in MLIR integration tests for vector dialect (NFC) --- mlir/test/Integration/Dialect/Vector/CPU/test-compress.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-expand.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-gather.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-maskedload.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-maskedstore.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-scatter.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-3d.mlir | 1 + mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read.mlir | 3 +++ mlir/test/Integration/Dialect/Vector/CPU/test-transfer-write.mlir | 1 + 9 files changed, 11 insertions(+) diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-compress.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-compress.mlir index 9c72014..fecf3fc 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-compress.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-compress.mlir @@ -100,5 +100,6 @@ func @entry() { call @printmem16(%A) : (memref) -> () // CHECK-NEXT: ( 0, 1, 2, 3, 11, 13, 15, 7, 0, 1, 2, 3, 12, 13, 14, 15 ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-expand.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-expand.mlir index b1cc01a..74c7028 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-expand.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-expand.mlir @@ -93,5 +93,6 @@ func @entry() { vector.print %e7 : vector<16xf32> // CHECK-NEXT: ( 8, 9, 10, 11, -7, -7, -7, -7, -7, -7, -7, -7, -7, -7, -7, -7 ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-gather.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-gather.mlir index eddc4ac..ff8a899 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-gather.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-gather.mlir @@ -86,5 +86,6 @@ func @entry() { vector.print %g5 : vector<8xf32> // CHECK: ( 0, 6, 1, 3, 5, 4, 9, 2 ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-maskedload.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-maskedload.mlir index c35ae8d..ebb25a2 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-maskedload.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-maskedload.mlir @@ -75,6 +75,7 @@ func @entry() { vector.print %l5 : vector<16xf32> // CHECK: ( 8, 9, 10, 11, 12, 13, 14, 15, -7, -7, -7, -7, -7, -7, -7, -7 ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-maskedstore.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-maskedstore.mlir index ef89491..6466771 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-maskedstore.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-maskedstore.mlir @@ -99,5 +99,6 @@ func @entry() { call @printmem16(%A) : (memref) -> () // CHECK: ( 0, 1, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 4, 5, 6, 7 ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-scatter.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-scatter.mlir index 0b30c7e..2e570dc 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-scatter.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-scatter.mlir @@ -114,5 +114,6 @@ func @entry() { call @printmem8(%A) : (memref) -> () // CHECK: ( 1, 2, 4, 7, 5, 6, 3, 0 ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-3d.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-3d.mlir index 2842096..6564552 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-3d.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-3d.mlir @@ -144,5 +144,6 @@ func @entry() { : (memref, index, index, index, index) -> () // CHECK: ( ( ( -42, -42, -42 ), ( -42, -42, -42 ), ( -42, -42, -42 ), ( -42, -42, -42 ), ( -42, -42, -42 ) ), ( ( 20, 20, 20 ), ( 20, 20, 20 ), ( 20, 20, 20 ), ( 20, 20, 20 ), ( 20, 20, 20 ) ) ) + memref.dealloc %A : memref return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read.mlir index 0817527..25dd32b5 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read.mlir @@ -82,6 +82,9 @@ func @entry() { call @transfer_read_inbounds_4(%A, %c1) : (memref, index) -> () // Read in-bounds with mask. call @transfer_read_mask_inbounds_4(%A, %c1) : (memref, index) -> () + + memref.dealloc %A : memref + return } diff --git a/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-write.mlir b/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-write.mlir index 6cca0cc..8cad0c9 100644 --- a/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-write.mlir +++ b/mlir/test/Integration/Dialect/Vector/CPU/test-transfer-write.mlir @@ -89,6 +89,7 @@ func @entry() { %6 = call @transfer_read_1d(%A) : (memref) -> (vector<32xf32>) vector.print %6 : vector<32xf32> + memref.dealloc %A : memref return } -- 2.7.4