0e39b1348e
This commit builds on and completes the work done in 9f6c632ecda08bfff76b798c46d5d7cfde57b5e9 to eliminate the need for a separate mlir-spirv-cpu-runner binary. Since the MLIR processing is already done outside this runner, the only real difference between it and the mlir-cpu-runner is the final linking step between the nested LLVM IR modules. By moving this step into mlir-cpu-runner behind a new command-line flag (`--link-nested-modules`), this commit is able to completely remove the runner component of the mlir-spirv-cpu-runner. The runtime libraries and the tests are moved and renamed to fit into the Execution Engine and Integration tests, following the model of the similar migration done for the CUDA Runner in D97463.
49 lines
1.6 KiB
C++
49 lines
1.6 KiB
C++
//===- SpirvCpuRuntimeWrappers.cpp - Runner testing library -===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// A small library for SPIR-V cpu runner testing.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/ExecutionEngine/CRunnerUtils.h"
|
|
|
|
#ifdef _WIN32
|
|
#define EXPORT __declspec(dllexport)
|
|
#else
|
|
#define EXPORT __attribute__((visibility("default")))
|
|
#endif
|
|
|
|
// NOLINTBEGIN(*-identifier-naming)
|
|
|
|
extern "C" EXPORT void
|
|
_mlir_ciface_fillI32Buffer(StridedMemRefType<int32_t, 1> *mem_ref,
|
|
int32_t value) {
|
|
std::fill_n(mem_ref->basePtr, mem_ref->sizes[0], value);
|
|
}
|
|
|
|
extern "C" EXPORT void
|
|
_mlir_ciface_fillF32Buffer1D(StridedMemRefType<float, 1> *mem_ref,
|
|
float value) {
|
|
std::fill_n(mem_ref->basePtr, mem_ref->sizes[0], value);
|
|
}
|
|
|
|
extern "C" EXPORT void
|
|
_mlir_ciface_fillF32Buffer2D(StridedMemRefType<float, 2> *mem_ref,
|
|
float value) {
|
|
std::fill_n(mem_ref->basePtr, mem_ref->sizes[0] * mem_ref->sizes[1], value);
|
|
}
|
|
|
|
extern "C" EXPORT void
|
|
_mlir_ciface_fillF32Buffer3D(StridedMemRefType<float, 3> *mem_ref,
|
|
float value) {
|
|
std::fill_n(mem_ref->basePtr,
|
|
mem_ref->sizes[0] * mem_ref->sizes[1] * mem_ref->sizes[2], value);
|
|
}
|
|
|
|
// NOLINTEND(*-identifier-naming)
|