pyopencl -如何使用泛型类型？

import pyopencl as cl
import numpy as np

# 创建上下文
ctx = cl.create_some_context()

# 创建命令队列
queue = cl.CommandQueue(ctx)

# 定义输入数据
input_data = np.array([1, 2, 3, 4, 5], dtype=np.float32)

# 创建输入缓冲区
input_buffer = cl.Buffer(ctx, cl.mem_flags.READ_ONLY | cl.mem_flags.COPY_HOST_PTR, hostbuf=input_data)

# 定义输出数据
output_data = np.empty_like(input_data)

# 创建输出缓冲区
output_buffer = cl.Buffer(ctx, cl.mem_flags.WRITE_ONLY, size=output_data.nbytes)

# 创建内核程序
program = cl.Program(ctx, """
    __kernel void square(__global const float* input, __global float* output) {
        int i = get_global_id(0);
        output[i] = input[i] * input[i];
    }
""").build()

# 设置内核参数
program.square(queue, input_data.shape, None, input_buffer, output_buffer)

# 读取输出数据
cl.enqueue_copy(queue, output_data, output_buffer)

# 打印输出数据
print("Output:", output_data)