Developer Story: How We Ported oneDNN to Fugaku with Arm
Fujitsu optimized and ported the oneDNN DL process library software for the Armv8-A instruction set so that it can be run at high speed on the Fugaku supercomputer. Senior Research for the Fujitsu Laboratories Platform Innovation Project, Kentaro Kawakami, shares how they introduced the JIT assembler, a key technology required for porting oneDNN, as well as the development history which accelerates port development. Fujitsu managed to make full use of Arm SVE architecture, resulting in improving performance 9.2 times in training and 7.8 times in inference. The assembler they developed has been officially incorporated into the main branch of oneDNN along with the source code optimized for the Armv8-A instruction set.