是否可以将shorts数组传递到CUDA内核

#include <cuda.h>
#include <stdio.h>

// CUDA内核函数
__global__ void myKernel(short* shortsArray)
{
    // 在这里对shortsArray进行操作
    // ...
}

int main()
{
    // 定义并初始化shorts数组
    short shorts[] = {1, 2, 3, 4, 5};

    // 定义设备指针和数组大小
    short* devShorts;
    int arraySize = sizeof(shorts);

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

    // 将数据从主机内存复制到设备内存
    cudaMemcpy(devShorts, shorts, arraySize, cudaMemcpyHostToDevice);

    // 调用CUDA内核函数
    myKernel<<<1, 1>>>(devShorts);

    // 在这里可以将数据从设备内存复制回主机内存
    // ...

    // 释放设备内存
    cudaFree(devShorts);

    return 0;
}