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 extern "C" void
  16 _mlir_ciface_fillI32Buffer(StridedMemRefType<int32_t, 1> *mem_ref,
  17                            int32_t value) {
  18   std::fill_n(mem_ref->basePtr, mem_ref->sizes[0], value);
  19 }
  20
  21 extern "C" void
  22 _mlir_ciface_fillF32Buffer1D(StridedMemRefType<float, 1> *mem_ref,
  23                              float value) {
  24   std::fill_n(mem_ref->basePtr, mem_ref->sizes[0], value);
  25 }
  26
  27 extern "C" void
  28 _mlir_ciface_fillF32Buffer2D(StridedMemRefType<float, 2> *mem_ref,
  29                              float value) {
  30   std::fill_n(mem_ref->basePtr, mem_ref->sizes[0] * mem_ref->sizes[1], value);
  31 }
  32
  33 extern "C" void
  34 _mlir_ciface_fillF32Buffer3D(StridedMemRefType<float, 3> *mem_ref,
  35                              float value) {
  36   std::fill_n(mem_ref->basePtr,
  37               mem_ref->sizes[0] * mem_ref->sizes[1] * mem_ref->sizes[2], value);
  38 }