Fix memory leak in mlir-cpu-runner/sgemm_naive_codegen.mlir (NFC)

joker-eph · joker-eph · commit fe48ecb047fa · 2021-10-02T23:07:49.000Z
diff --git a/mlir/test/mlir-cpu-runner/sgemm_naive_codegen.mlir b/mlir/test/mlir-cpu-runner/sgemm_naive_codegen.mlir
@@ -42,6 +42,9 @@ func @main() {
   %flops = divf %num_flops_f, %t : f64
   call @print_flops(%flops) : (f64) -> ()
 
+  memref.dealloc %A : memref<16x16xf32>
+  memref.dealloc %B : memref<16x16xf32>
+  memref.dealloc %C : memref<16x16xf32>
   return
 }
 // CHECK: 17
