关于任意Size如何处理?比如实现VecAdd(float * z, const float * x, const float *z, const int size)接口
收藏回复举报
关于任意Size如何处理?比如实现VecAdd(float * z, const float * x, const float *z, const int size)接口
t('forum.solved') 已解决
发表于2024-07-04 18:40:37
0 查看

看了不少sample,基本都是固定shape(sample code建议搞点真实aclnn API的真实实现的样例,而不仅仅是一个sample code)

1. kernel function参数除了指针,是否接受其它参数

看文档写了支持仅支持入参为指针或C/C++内置数据类型, 

比如,__global__ __aicore__ void AddKernel(GM_ADDR z, GM_ADDR x,GM_ADDR y, const int size)这样的

但sample code没看到过这样的  

2. 假如计算1024长度为一片,最后尾部的,

int valid_size = size - last_block_idx * 1024; 

DataCopy(local_tensor, global_tensor, valid_size) 这样处理就可以?

我要发帖子