27ed1f99e2
This patch extends the conformance testing infrastructure to support two new providers of math function implementations for GPUs: CUDA Math (`cuda-math`) and HIP Math (`hip-math`).
36 lines
1.2 KiB
C++
36 lines
1.2 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// This file contains the definition of the runKernelBody, a template helper
|
|
/// that executes the per-thread logic of a math function's kernel wrapper.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef CONFORMANCE_DEVICE_CODE_KERNELRUNNER_HPP
|
|
#define CONFORMANCE_DEVICE_CODE_KERNELRUNNER_HPP
|
|
|
|
#include <gpuintrin.h>
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
namespace kernels {
|
|
|
|
template <auto Func, typename OutType, typename... InTypes>
|
|
void runKernelBody(size_t NumElements, OutType *Out, const InTypes *...Ins) {
|
|
uint32_t Index =
|
|
__gpu_num_threads_x() * __gpu_block_id_x() + __gpu_thread_id_x();
|
|
|
|
if (Index < NumElements) {
|
|
Out[Index] = Func(Ins[Index]...);
|
|
}
|
|
}
|
|
} // namespace kernels
|
|
|
|
#endif // CONFORMANCE_DEVICE_CODE_KERNELRUNNER_HPP
|