基于GPU的线程排布方法、装置、设备以及存储介质

基本信息

申请号 CN202110835793.9 申请日 -
公开(公告)号 CN113641470A 公开(公告)日 2021-11-12
申请公布号 CN113641470A 申请公布日 2021-11-12
分类号 G06F9/48(2006.01)I;G06F9/50(2006.01)I 分类 计算;推算;计数;
发明人 赵源;黎雷生;朱鹏飞 申请(专利权)人 曙光云计算集团有限公司
代理机构 北京清亦华知识产权代理事务所(普通合伙) 代理人 黄玉霞
地址 100193北京市海淀区东北旺西路8号院36号楼5层
法律状态 -

摘要

摘要 本发明公开了一种基于GPU的线程排布方法、装置、设备以及存储介质,所述方法包括:获取所需要的线程;将所述线程分配到一维网格中;通过核函数将待计算的数组映射到所述线程上。本发明的技术方案,通过优化线程的排布,将线程映射到一维网格中,使得线程可以按需分配,提高线程的利用率,可以使得很多计算不是非常复杂的核函数的性能得到一定程度的提升,且不需要修改核函数中的计算代码,只需要修改元素的映射代码,既可以降低操作难度也可以减少工作量;同时,本发明提供的映射方法,不仅可以计算三维数组,还可以很轻松的实现维数大于3的数组元素的并行计算,应用范围广泛。