Fichiers
rocm-systems/samples/0_Intro/module_api/vcpy_kernel.cpp
T

15 lignes
241 B
C++
Brut Vue normale Historique

#include<hip_runtime.h>
#ifdef __HIP_PLATFORM_NVCC__
#define EXTERN_C extern "C"
#else
#define EXTERN_C
#endif
EXTERN_C __global__ void hello_world(hipLaunchParm lp, float *a, float *b)
{
int tx = hipThreadIdx_x;
b[tx] = a[tx];
}