您好,这里需要调用cnrtGetKernelParamsBuffer,例如:
cnrtKernelParamsBuffer_t params;
cnrtGetKernelParamsBuffer(¶ms);
cnrtKernelParamsBufferAddParam(params, &mlu_input, sizeof(half*));
请登录后评论