OpenCL TensorFlow implementation method and device, electronic equipment and storage medium

• Main Authors: CHEN YUQIAO, SUN YUFEI, GUO QIANG, SUI YICHENG, SHI CHANGQING, CHEN RUI, ZHANG YUZHE, ZHANG YUZHI, CHENG DAGUO
• Format: Patent
• Language: Chinese; English
• Published: 01.11.2022
• Subjects: CALCULATING; COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS; COMPUTING; COUNTING; ELECTRIC DIGITAL DATA PROCESSING; PHYSICS

The invention provides an implementation method and device of OpenCL TensorFlow, electronic equipment and a storage medium. The method comprises the following steps of: realizing OpenCL Device support of a TensorFlow equipment layer on the basis of an OpenCL programming standard; according to a calculation function of a preset kernel in TensorFlow, an OpenCL kernel and an OpenCL operator corresponding to the calculation function are realized based on an OpenCL programming standard; the method comprises the following steps of: realizing an OpenCLEleft-will operator library on the basis of an OpenCL (Open Computing Language) programming standard; carrying out OpenCL standard adaptation on a calculation acceleration library required by the task; an API interface of the first OpenCL calculation acceleration library is used for achieving integrated calling; according to the method, the support of the StreamExecutor on the OpenCL is realized, and the integrated calling of the second OpenCL calculation acceleration