ROCmKernel/src/DeviceAdditionKernel.hip.cc

0001 #include <cstddef>
0002
0003 #include <hip/hip_runtime.h>
0004
0005 #include "HeterogeneousTest/ROCmDevice/interface/DeviceAddition.h"
0006 #include "HeterogeneousTest/ROCmKernel/interface/DeviceAdditionKernel.h"
0007
0008 namespace cms::rocmtest {
0009
0010   __global__ void kernel_add_vectors_f(const float* __restrict__ in1,
0011                                        const float* __restrict__ in2,
0012                                        float* __restrict__ out,
0013                                        size_t size) {
0014     add_vectors_f(in1, in2, out, size);
0015   }
0016
0017   __global__ void kernel_add_vectors_d(const double* __restrict__ in1,
0018                                        const double* __restrict__ in2,
0019                                        double* __restrict__ out,
0020                                        size_t size) {
0021     add_vectors_d(in1, in2, out, size);
0022   }
0023
0024 }  // namespace cms::rocmtest