Files
learning-hip/kernels/kernels.hip

11 lines
210 B
Plaintext

#include <hip/hip_runtime.h>
extern "C" __global__
void saxpy(float* y, const float* x, float a, int n) {
int i = blockIdx.x * blockDim.x + threadIdx.x;
if (i < n) {
y[i] += a * x[i];
}
}