在CUDA中测试无穷大

#include <stdio.h>
#include <math.h>

__global__ void checkInfinity(float* data, int size) {
    int tid = blockIdx.x * blockDim.x + threadIdx.x;
    if (tid < size) {
        if (isinf(data[tid])) {
            printf("Element at index %d is infinity\n", tid);
        }
    }
}

int main() {
    int size = 10;
    float data[] = {1.0, 2.0, INFINITY, 4.0, -INFINITY, 6.0, 7.0, 8.0, 9.0, 10.0};
    float* d_data;

    cudaMalloc((void**)&d_data, size * sizeof(float));
    cudaMemcpy(d_data, data, size * sizeof(float), cudaMemcpyHostToDevice);

    int blockSize = 256;
    int numBlocks = (size + blockSize - 1) / blockSize;

    checkInfinity<<<numBlocks, blockSize>>>(d_data, size);

    cudaFree(d_data);

    return 0;
}