ROCmKernel/plugins/ROCmTestKernelAdditionAlgo.hip.cc

0001 #include <cstddef>
0002
0003 #include <hip/hip_runtime.h>
0004
0005 #include "HeterogeneousTest/ROCmKernel/interface/DeviceAdditionKernel.h"
0006 #include "HeterogeneousCore/ROCmUtilities/interface/hipCheck.h"
0007
0008 #include "ROCmTestKernelAdditionAlgo.h"
0009
0010 namespace HeterogeneousTestROCmKernelPlugins {
0011
0012   void wrapper_add_vectors_f(const float* __restrict__ in1,
0013                              const float* __restrict__ in2,
0014                              float* __restrict__ out,
0015                              size_t size) {
0016     cms::rocmtest::kernel_add_vectors_f<<<32, 32>>>(in1, in2, out, size);
0017     hipCheck(hipGetLastError());
0018   }
0019
0020 }  // namespace HeterogeneousTestROCmKernelPlugins