1. 您所在位置：
  2. 网站首页
  3. 文档分类
  4. 行业资料
  5. 国内专利

GPU图神经网络优化方法及装置.pdfVIP

  • 8
  • 0
  • 约1.19万字
  • 约 13页
  • 2023-06-06 发布于四川
  • 举报

GPU图神经网络优化方法及装置.pdf

    提供GPU图神经网络优化方法和计算机可读介质，优化方法包括：对于GPU图神经网络模型定义，生成包括张量和操作的计算图；针对所述计算图，得到多个等价计算图；比较各个计算图的计算量，选取计算量最小的计算图；对于选取的计算图，生成对应GPU代码。可以对于选取的计算图，分析得到其每个操作所需输入的数据可视范围，以及输出的数据可视范围；解决有依赖关系的操作的数据可视范围不匹配的问题；将数据可视已匹配的操作合并为同一个GPU核函数。本发明的图神经网络优化方法可以减少图神经网络中的算子之间的数据可视范围不匹配

    (19)中华人民共和国国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 112767230 A (43)申请公布日 2021.05.07 (21)申请号 202110222831.3 (22)申请日 2021.02.26 (71)申请人 清华大学 地址 10

    您可能关注的文档

    最近下载

    文档评论（0）

    1亿VIP精品文档

    更多 >

    相关文档

    更多 >