公开(公告)号：CN115268857A

公开(公告)日：2022-11-01

申请号：CN202210698399.X

申请日：2022-06-20

Applicant: 南开大学

Inventor： 孙羽菲 ,  陈锐 ,  石昌青 ,  郭强 ,  陈禹乔 ,  周建宇 ,  张玉志

IPC: G06F8/30 ,  G06F9/448 ,  G06N3/063

Abstract: 本公开公开了OpenCLDNN库的集成方法及装置、系统和电子设备，具体涉及人工智能技术领域，在StreamExecutor中实现OpenCL Platform，并注册到TensorFlow框架中；所述OpenCL Platform为根据指令进行创建、指定对应的设备Platform；在StreamExecutor中实现OpenCLExecutor，并注册到TensorFlow框架中；在StreamExecutor中的LibrarySupport层完成OpenCL DNN的注册和接入；在TensorFlow框架中将OpenCL DNN库中对应的算子与OpenCL设备绑定并进行注册；实现了OpenCL DNN库在TensorFlow框架中的集成，使得TensorFlow框架在OpenCL硬件设备上运行时可以对深度神经网络常用的算子进行计算加速。