• Complex
  • Title
  • Keyword
  • Abstract
  • Scholars
  • Journal
  • ISSN
  • Conference
搜索

Author:

Bao, Z. (Bao, Z..) | Li, Y. (Li, Y..) | Zhang, W. (Zhang, W..)

Indexed by:

EI Scopus

Abstract:

Generalized matrix multiplication (GEMM) has become the fundamental operator in deep learning. Operators of deep learning need to be transformed and optimized into executable code for specific hardware. However, existing deep learning frameworks rely on handwritten libraries that are not scalable for various hardware target. In this paper we introduce a compiler-level tuning algorithm based on Neighborhood Actor Advantage Critic (N-A2C) method for GEMM tiling optimization. Experiment results shows that our method achieves 8% and 20% computational time savings of GEMM compared Xgboost and RNN methods when exploring 1% of the search space. © 2024 ACM.

Keyword:

Xgboost deep learning AI compiler random search reinforcement learning

Author Community:

  • [ 1 ] [Bao Z.]Faculty of Information Technology, Beijing University of Technology, Beijing, 100124, China
  • [ 2 ] [Li Y.]Faculty of Information Technology, Beijing University of Technology, Beijing, 100124, China
  • [ 3 ] [Zhang W.]Faculty of Information Technology, Beijing University of Technology, Beijing, 100124, China

Reprint Author's Address:

Email:

Show more details

Related Keywords:

Related Article:

Source :

Year: 2024

Page: 79-83

Language: English

Cited Count:

WoS CC Cited Count:

SCOPUS Cited Count:

ESI Highly Cited Papers on the List: 0 Unfold All

WanFang Cited Count:

Chinese Cited Count:

30 Days PV: 7

Affiliated Colleges:

Online/Total:986/10805754
Address:BJUT Library(100 Pingleyuan,Chaoyang District,Beijing 100124, China Post Code:100124) Contact Us:010-67392185
Copyright:BJUT Library Technical Support:Beijing Aegean Software Co., Ltd.