CUDA根据条件填充较小的数组

#include <stdio.h>
#include <cuda_runtime.h>

#define ARRAY_SIZE 1024

__global__ void fillArray(int* array, int value, int condition)
{
    int index = blockIdx.x * blockDim.x + threadIdx.x;
    if (index < ARRAY_SIZE && index < condition)
    {
        array[index] = value;
    }
}

int main()
{
    int* d_array;
    int arraySize = ARRAY_SIZE * sizeof(int);

    // 在设备上分配内存
    cudaMalloc((void**)&d_array, arraySize);

    // 填充数组为0
    cudaMemset(d_array, 0, arraySize);

    // 根据条件填充数组为1
    fillArray<<<1, ARRAY_SIZE>>>(d_array, 1, 256);

    // 将结果从设备复制回主机
    int h_array[ARRAY_SIZE];
    cudaMemcpy(h_array, d_array, arraySize, cudaMemcpyDeviceToHost);

    // 打印结果
    for (int i = 0; i < ARRAY_SIZE; i++)
    {
        printf("%d ", h_array[i]);
    }

    // 释放设备上的内存
    cudaFree(d_array);

    return 0;
}