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
_mlir_ciface_fillI32Buffer(StridedMemRefType<int32_t,1> * mem_ref,int32_t value)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
_mlir_ciface_fillF32Buffer1D(StridedMemRefType<float,1> * mem_ref,float value)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
_mlir_ciface_fillF32Buffer2D(StridedMemRefType<float,2> * mem_ref,float value)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
_mlir_ciface_fillF32Buffer3D(StridedMemRefType<float,3> * mem_ref,float value)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)
43