From 82a84557186f1a3ecb85dfc1c5f5b5d426070a12 Mon Sep 17 00:00:00 2001
From: Taiqi Zheng <56971484+taiqzheng@users.noreply.github.com>
Date: Thu, 8 Jun 2023 08:48:20 +0800
Subject: [PATCH] [tests] Add rountrip tests for DAP and DIP operations. (#157)

- [DAP] iir & fir
- [DIP] corr2d & resize2d & rotate2d
---
 tests/Dialect/DAP/filters_roundtrip.mlir      | 13 ++++++++
 .../DIP/image_processing_roundtrip.mlir       | 31 +++++++++++++++++++
 2 files changed, 44 insertions(+)
 create mode 100644 tests/Dialect/DAP/filters_roundtrip.mlir
 create mode 100644 tests/Dialect/DIP/image_processing_roundtrip.mlir

diff --git a/tests/Dialect/DAP/filters_roundtrip.mlir b/tests/Dialect/DAP/filters_roundtrip.mlir
new file mode 100644
index 000000000..6f4698647
--- /dev/null
+++ b/tests/Dialect/DAP/filters_roundtrip.mlir
@@ -0,0 +1,13 @@
+// RUN: buddy-opt -verify-diagnostics %s | buddy-opt | FileCheck %s
+
+func.func @buddy_iir(%in : memref<?xf32>, %filter : memref<?x?xf32>, %out : memref<?xf32>) -> () {
+  // CHECK: dap.iir {{.*}} : memref<?xf32>, memref<?x?xf32>, memref<?xf32>
+  dap.iir %in, %filter, %out : memref<?xf32>, memref<?x?xf32>, memref<?xf32>
+  return
+} 
+
+func.func @buddy_fir(%in : memref<?xf32>, %filter : memref<?xf32>, %out : memref<?xf32>) -> () {
+  // CHECK: dap.fir {{.*}} : memref<?xf32>, memref<?xf32>, memref<?xf32>
+  dap.fir %in, %filter, %out : memref<?xf32>, memref<?xf32>, memref<?xf32>
+  return
+}
diff --git a/tests/Dialect/DIP/image_processing_roundtrip.mlir b/tests/Dialect/DIP/image_processing_roundtrip.mlir
new file mode 100644
index 000000000..768962087
--- /dev/null
+++ b/tests/Dialect/DIP/image_processing_roundtrip.mlir
@@ -0,0 +1,31 @@
+// RUN: buddy-opt -verify-diagnostics %s | buddy-opt | FileCheck %s
+
+func.func @buddy_corr2d_CONSTANT_PADDING(%input : memref<?x?xf32>, %identity : memref<?x?xf32>, %output : memref<?x?xf32>, %kernelAnchorX : index, %kernelAnchorY : index, %c : f32) -> () {
+  // CHECK: dip.corr_2d <CONSTANT_PADDING>{{.*}} : memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>, index, index, f32
+  dip.corr_2d <CONSTANT_PADDING> %input, %identity, %output, %kernelAnchorX, %kernelAnchorY, %c : memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>, index, index, f32
+  return
+}
+
+func.func @buddy_corr2d_REPLICATE_PADDING(%input : memref<?x?xf32>, %identity : memref<?x?xf32>, %output : memref<?x?xf32>, %kernelAnchorX : index, %kernelAnchorY : index, %c : f32) -> () {
+  // CHECK: dip.corr_2d <REPLICATE_PADDING>{{.*}} : memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>, index, index, f32
+  dip.corr_2d <REPLICATE_PADDING> %input, %identity, %output, %kernelAnchorX, %kernelAnchorY, %c : memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>, index, index, f32
+  return
+}
+
+func.func @buddy_resize2d_NEAREST_NEIGHBOUR_INTERPOLATION(%input : memref<?x?xf32>, %horizontal_scaling_factor : f32, %vertical_scaling_factor : f32, %output : memref<?x?xf32>) -> () {
+  // CHECK: dip.resize_2d NEAREST_NEIGHBOUR_INTERPOLATION{{.*}} : memref<?x?xf32>, f32, f32, memref<?x?xf32>
+  dip.resize_2d NEAREST_NEIGHBOUR_INTERPOLATION %input, %horizontal_scaling_factor, %vertical_scaling_factor, %output : memref<?x?xf32>, f32, f32, memref<?x?xf32>
+  return
+}
+
+func.func @buddy_resize2d_BILINEAR_INTERPOLATION(%input : memref<?x?xf32>, %horizontal_scaling_factor : f32, %vertical_scaling_factor : f32, %output : memref<?x?xf32>) -> () {
+  // CHECK: dip.resize_2d BILINEAR_INTERPOLATION{{.*}} : memref<?x?xf32>, f32, f32, memref<?x?xf32>
+  dip.resize_2d BILINEAR_INTERPOLATION %input, %horizontal_scaling_factor, %vertical_scaling_factor, %output : memref<?x?xf32>, f32, f32, memref<?x?xf32>
+  return
+}
+
+func.func @buddy_rotate2d(%input : memref<?x?xf32>, %angle : f32, %output : memref<?x?xf32>) -> () {
+  // CHECK: dip.rotate_2d {{.*}} : memref<?x?xf32>, f32, memref<?x?xf32>
+  dip.rotate_2d %input, %angle, %output : memref<?x?xf32>, f32, memref<?x?xf32>
+  return
+}