写一段c语言测试gpu性能的程序
时间: 2024-03-17 11:39:45 浏览: 135
一个小程序测试cpu
好的,下面是一个简单的使用CUDA测试GPU性能的C语言程序,它会计算一个大矩阵的乘积。请注意,这个程序仅供参考,实际应用中需要根据具体情况进行修改和优化。
```c
#include <stdio.h>
#include <stdlib.h>
#include <cuda_runtime.h>
#define N 1024 // 矩阵大小
__global__ void matrixMul(float *a, float *b, float *c, int n)
{
int row = blockIdx.y * blockDim.y + threadIdx.y;
int col = blockIdx.x * blockDim.x + threadIdx.x;
if (row < n && col < n) {
float sum = 0.0f;
for (int i = 0; i < n; i++) {
sum += a[row * n + i] * b[i * n + col];
}
c[row * n + col] = sum;
}
}
int main()
{
float *h_a, *h_b, *h_c;
float *d_a, *d_b, *d_c;
int size = N * N * sizeof(float);
// 分配内存
h_a = (float*)malloc(size);
h_b = (float*)malloc(size);
h_c = (float*)malloc(size);
cudaMalloc((void**)&d_a, size);
cudaMalloc((void**)&d_b, size);
cudaMalloc((void**)&d_c, size);
// 初始化数据
for (int i = 0; i < N * N; i++) {
h_a[i] = rand() / (float)RAND_MAX;
h_b[i] = rand() / (float)RAND_MAX;
}
// 将数据从主机内存拷贝到设备内存
cudaMemcpy(d_a, h_a, size, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, h_b, size, cudaMemcpyHostToDevice);
// 调用CUDA核函数
dim3 block(16, 16);
dim3 grid((N + block.x - 1) / block.x, (N + block.y - 1) / block.y);
matrixMul<<<grid, block>>>(d_a, d_b, d_c, N);
// 将数据从设备内存拷贝到主机内存
cudaMemcpy(h_c, d_c, size, cudaMemcpyDeviceToHost);
// 验证结果
float eps = 1e-6;
for (int i = 0; i < N * N; i++) {
if (abs(h_c[i] - (N / 2)) > eps) {
printf("Error: incorrect results\n");
break;
}
}
// 释放内存
free(h_a);
free(h_b);
free(h_c);
cudaFree(d_a);
cudaFree(d_b);
cudaFree(d_c);
return 0;
}
```
这个程序会生成两个随机的N×N的矩阵,然后计算它们的乘积,并将结果存储在第三个矩阵中。程序会将数据从主机内存拷贝到设备内存,然后调用CUDA核函数进行计算,最后将结果从设备内存拷贝回主机内存进行验证。您可以根据需要修改矩阵大小和块大小等参数,以及优化程序代码来提高程序的性能。
阅读全文