[DIP] Add benchmark for Corr2D.

benchmarks/ImageProcessing/CMakeLists.txt

@@ -2,22 +2,52 @@ set(BUDDY_OPT_STRIP_MINING 256)
 set(BUDDY_OPT_ATTR avx512f)
 set(LLVM_MLIR_BINARY_DIR ${BUDDY_OPT_BUILD_DIR}/../llvm/build/bin)
 
+#-------------------------------------------------------------------------------
+# MLIR Linalg Dialect Conv2D Operation + CB Algorithm
+#-------------------------------------------------------------------------------
+
 add_custom_command(OUTPUT conv2d.o
   COMMAND ${BUDDY_OPT_BUILD_DIR}/bin/buddy-opt ${BUDDY_SOURCE_DIR}/benchmarks/ImageProcessing/Conv2D.mlir -conv-vectorization="strip-mining=${BUDDY_OPT_STRIP_MINING}" -lower-affine -convert-scf-to-std -convert-vector-to-llvm -convert-memref-to-llvm -convert-std-to-llvm='emit-c-wrappers=1' -reconcile-unrealized-casts | 
           ${LLVM_MLIR_BINARY_DIR}/mlir-translate --mlir-to-llvmir |
           ${LLVM_MLIR_BINARY_DIR}/llc -mtriple=x86_64-unknown-linux-gnu -mattr=${BUDDY_OPT_ATTR} --filetype=obj -o ${BUDDY_BINARY_DIR}/../benchmarks/ImageProcessing/conv2d.o
 )
-
 add_library(Conv2D STATIC conv2d.o)
-
 set_target_properties(Conv2D PROPERTIES LINKER_LANGUAGE CXX)
 
-add_executable(image-processing-benchmark Main.cpp OpenCVBenchmark.cpp BuddyBenchmark.cpp)
+#-------------------------------------------------------------------------------
+# Buddy DIP Dialect Corr2D Operation + CB Algorithm
+#-------------------------------------------------------------------------------
+
+add_custom_command(OUTPUT corr2d.o
+  COMMAND ${BUDDY_OPT_BUILD_DIR}/bin/buddy-opt 
+          ${BUDDY_SOURCE_DIR}/benchmarks/ImageProcessing/Corr2D.mlir 
+            -lower-dip -lower-affine -convert-scf-to-std -convert-vector-to-llvm 
+            -convert-memref-to-llvm -convert-std-to-llvm='emit-c-wrappers=1' 
+            -reconcile-unrealized-casts | 
+          ${LLVM_MLIR_BINARY_DIR}/mlir-translate --mlir-to-llvmir |
+          ${LLVM_MLIR_BINARY_DIR}/llc -mtriple=x86_64-unknown-linux-gnu 
+            -mattr=${BUDDY_OPT_ATTR} --filetype=obj 
+            -o ${BUDDY_BINARY_DIR}/../benchmarks/ImageProcessing/corr2d.o
+)
+add_library(Corr2D STATIC corr2d.o)
+set_target_properties(Corr2D PROPERTIES LINKER_LANGUAGE CXX)
+
+#-------------------------------------------------------------------------------
+# Image Processing Benchmark Target
+#-------------------------------------------------------------------------------
+
+add_executable(image-processing-benchmark
+  Main.cpp 
+  OpenCVBenchmark.cpp
+  BuddyBenchmark.cpp
+  Corr2DBenchmark.cpp
+  )
 
 target_link_libraries(image-processing-benchmark 
   GoogleBenchmark
   ${OpenCV_LIBS}
   Conv2D
+  Corr2D
   Container
   PNGImage
   )

benchmarks/ImageProcessing/Corr2D.mlir

+func @corr_2d(%inputImage : memref<?x?xf32>, %kernel : memref<?x?xf32>, 
+              %outputImage : memref<?x?xf32>, %centerX : index, 
+              %centerY : index, %boundaryOption : index) {
+  dip.corr_2d %inputImage, %kernel, %outputImage, %centerX, %centerY, 
+              %boundaryOption : memref<?x?xf32>, memref<?x?xf32>, 
+                                memref<?x?xf32>, index, index, index
+  return
+}
\ No newline at end of file

benchmarks/ImageProcessing/Corr2DBenchmark.cpp

+//===- Corr2DBenchmark.cpp ------------------------------------------------===//
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the benchmark for Corr2D operation.
+//
+//===----------------------------------------------------------------------===//
+
+#include "ImageProcessing/Kernels.h"
+#include "Utils/Container.h"
+#include <benchmark/benchmark.h>
+#include <opencv2/opencv.hpp>
+
+using namespace cv;
+using namespace std;
+
+// Declare the conv2d C interface.
+extern "C" {
+void _mlir_ciface_corr_2d(MemRef<float, 2> *inputCorr2D,
+                          MemRef<float, 2> *kernelCorr2D,
+                          MemRef<float, 2> *outputCorr2D, unsigned int centerX,
+                          unsigned int centerY, int boundaryOption);
+}
+
+// Read input image.
+Mat inputImageCorr2D = imread(
+    "../../benchmarks/ImageProcessing/Images/YuTu.png", IMREAD_GRAYSCALE);
+
+// Define the kernel size.
+int kernelRowsCorr2D = laplacianKernelRows;
+int kernelColsCorr2D = laplacianKernelCols;
+
+// Define the output size.
+int outputRowsCorr2D = inputImageCorr2D.rows;
+int outputColsCorr2D = inputImageCorr2D.cols;
+
+// Define sizes of input, kernel, and output.
+intptr_t sizesInputCorr2D[2] = {inputImageCorr2D.rows, inputImageCorr2D.cols};
+intptr_t sizesKernelCorr2D[2] = {kernelRowsCorr2D, kernelColsCorr2D};
+intptr_t sizesOutputCorr2D[2] = {outputRowsCorr2D, outputColsCorr2D};
+
+// Define the MemRef descriptor for input, kernel, and output.
+MemRef<float, 2> inputCorr2D(inputImageCorr2D, sizesInputCorr2D);
+MemRef<float, 2> kernelCorr2D(laplacianKernelAlign, sizesKernelCorr2D);
+MemRef<float, 2> outputCorr2D(sizesOutputCorr2D);
+
+static void BM_Corr2D(benchmark::State &state) {
+  for (auto _ : state) {
+    for (int i = 0; i < state.range(0); ++i) {
+      _mlir_ciface_corr_2d(&inputCorr2D, &kernelCorr2D, &outputCorr2D,
+                           1 /* Center X */, 1 /* Center Y */,
+                           0 /* Boundary Option */);
+    }
+  }
+}
+
+// Register benchmarking function with different arguments.
+BENCHMARK(BM_Corr2D)->Arg(1);
+BENCHMARK(BM_Corr2D)->Arg(2);
+BENCHMARK(BM_Corr2D)->Arg(4);
+BENCHMARK(BM_Corr2D)->Arg(8);
+BENCHMARK(BM_Corr2D)->Arg(16);
+
+// Generate result image.
+void generateResultCorr2D() {
+  // Define the MemRef descriptor for input, kernel, and output.
+  MemRef<float, 2> input(inputImageCorr2D, sizesInputCorr2D);
+  MemRef<float, 2> kernel(laplacianKernelAlign, sizesKernelCorr2D);
+  MemRef<float, 2> output(sizesOutputCorr2D);
+  // Run the 2D correlation.
+  _mlir_ciface_corr_2d(&input, &kernel, &output, 1 /* Center X */,
+                       1 /* Center Y */, 0 /* Boundary Option */);
+
+  // Define a cv::Mat with the output of the correlation.
+  Mat outputImage(outputRowsCorr2D, outputColsCorr2D, CV_32FC1,
+                  output.getData());
+
+  // Choose a PNG compression level
+  vector<int> compressionParams;
+  compressionParams.push_back(IMWRITE_PNG_COMPRESSION);
+  compressionParams.push_back(9);
+
+  // Write output to PNG.
+  bool result = false;
+  try {
+    result = imwrite("ResultCorr2D.png", outputImage, compressionParams);
+  } catch (const cv::Exception &ex) {
+    fprintf(stderr, "Exception converting image to PNG format: %s\n",
+            ex.what());
+  }
+  if (result)
+    cout << "Saved PNG file." << endl;
+  else
+    cout << "ERROR: Can't save PNG file." << endl;
+}

benchmarks/ImageProcessing/Main.cpp

@@ -20,10 +20,14 @@
 
 #include <benchmark/benchmark.h>
 
+void generateResultCorr2D();
+
 // Run benchmarks.
 int main(int argc, char **argv) {
   ::benchmark::Initialize(&argc, argv);
   ::benchmark::RunSpecifiedBenchmarks();
+  // Generate result image.
+  generateResultCorr2D();
 
   return 0;
 }