只是好奇如何解析这个.它看起来像一个模板化的方法,但我不理解三角括号’<<<<<这里发生了什么?
backward_scale_kernel<<<n,BLOCK>>>(x_norm,delta,batch,n,size,scale_updates);
对于上下文,“n”作为函数参数传入,我找不到BLOCK的定义或分配位置.
解决方法 我确信这已经有了很好的重复,但是<<< >>>您正在看到的装饰器是一个CUDA运行时API syntax extension,它允许指定CUDA内核调用的执行参数.完整的语法是
kernel_function<<<grID dimensions,block dimensions,dynamic shared memory,stream ID>>>( ....arguments....)
CUDA运行时API前端将此语法扩展为对编译器发出的样板的一对内联函数调用,以允许在运行时启动基础GPU内核.
总结以上是内存溢出为你收集整理的C/C++UDA中带有3个尖括号(<<<)的模板化方法是什么?全部内容,希望文章能够帮你解决C/C++UDA中带有3个尖括号(<<<)的模板化方法是什么?所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)