Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2023-10-25 09:50:30

0001 #ifndef HeterogeneousTest_ROCmDevice_interface_DeviceAddition_h
0002 #define HeterogeneousTest_ROCmDevice_interface_DeviceAddition_h
0003 
0004 #include <cstddef>
0005 
0006 #include <hip/hip_runtime.h>
0007 
0008 namespace cms::rocmtest {
0009 
0010   __device__ void add_vectors_f(const float* __restrict__ in1,
0011                                 const float* __restrict__ in2,
0012                                 float* __restrict__ out,
0013                                 size_t size);
0014 
0015   __device__ void add_vectors_d(const double* __restrict__ in1,
0016                                 const double* __restrict__ in2,
0017                                 double* __restrict__ out,
0018                                 size_t size);
0019 
0020 }  // namespace cms::rocmtest
0021 
0022 #endif  // HeterogeneousTest_ROCmDevice_interface_DeviceAddition_h