11 lines
210 B
Plaintext
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];
|
|
}
|
|
}
|
|
|