mlir/test/mlir-spirv-cpu-runner/mlir_test_spirv_cpu_runner_c_wrappers.cpp

   1 //===- mlir_test_spirv_cpu_runner_c_wrappers.cpp - Runner testing library -===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // A small library for SPIR-V cpu runner testing.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "mlir/ExecutionEngine/RunnerUtils.h"
  14
  15 // NOLINTBEGIN(*-identifier-naming)
  16
  17 extern "C" void
  18 _mlir_ciface_fillI32Buffer(StridedMemRefType<int32_t, 1> *mem_ref,
  19                            int32_t value) {
  20   std::fill_n(mem_ref->basePtr, mem_ref->sizes[0], value);
  21 }
  22
  23 extern "C" void
  24 _mlir_ciface_fillF32Buffer1D(StridedMemRefType<float, 1> *mem_ref,
  25                              float value) {
  26   std::fill_n(mem_ref->basePtr, mem_ref->sizes[0], value);
  27 }
  28
  29 extern "C" void
  30 _mlir_ciface_fillF32Buffer2D(StridedMemRefType<float, 2> *mem_ref,
  31                              float value) {
  32   std::fill_n(mem_ref->basePtr, mem_ref->sizes[0] * mem_ref->sizes[1], value);
  33 }
  34
  35 extern "C" void
  36 _mlir_ciface_fillF32Buffer3D(StridedMemRefType<float, 3> *mem_ref,
  37                              float value) {
  38   std::fill_n(mem_ref->basePtr,
  39               mem_ref->sizes[0] * mem_ref->sizes[1] * mem_ref->sizes[2], value);
  40 }
  41
  42 // NOLINTEND(*-identifier-naming)