ROCmWrapper/interface/DeviceAdditionWrapper.h

0001 #ifndef HeterogeneousTest_ROCmWrapper_interface_DeviceAdditionWrapper_h
0002 #define HeterogeneousTest_ROCmWrapper_interface_DeviceAdditionWrapper_h
0003
0004 #include <cstddef>
0005
0006 namespace cms::rocmtest {
0007
0008   void wrapper_add_vectors_f(const float* __restrict__ in1,
0009                              const float* __restrict__ in2,
0010                              float* __restrict__ out,
0011                              size_t size);
0012
0013   void wrapper_add_vectors_d(const double* __restrict__ in1,
0014                              const double* __restrict__ in2,
0015                              double* __restrict__ out,
0016                              size_t size);
0017
0018 }  // namespace cms::rocmtest
0019
0020 #endif  // HeterogeneousTest_ROCmWrapper_interface_DeviceAdditionWrapper_h