• Home
  • Search
  • Cost-Aware TVM (CAT) Tensorization for Modern Deep Learning Accelerators
  • https://doi.org/10.1109/iccd56317.2022.00058Copy DOI Icon

Cost-Aware TVM (CAT) Tensorization for Modern Deep Learning Accelerators

  • Oct 1, 2022
  • Yahang Hu +5 more
Show More
  • Abstract
  • Literature Map
  • References
  • Similar Papers
Abstract

TVM is proposed to efficiently deploy deep learning (DL) networks to diverse hardware devices. Tensorization in TVM enables designers to manually set up mappings between tensor computation patterns in DL networks and specific hardware tensor instructions, so as to fully exploit the potential of DL accelerators. However, with the development of modem DL accelerators, a much richer set of hardware instructions are being proposed. This leads to a one-to-many mapping between a tensor computation pattern and hardware instructions, making it a non-trivial task to choose a proper mapping. To this end, we propose Cost Aware TVM (CAT) tensorization. The key insight of CAT is that both tensor computation patterns and instructions share similar tensor features. Based on this insight, CAT first extracts the tensor features of tensor computation patterns and locates eligible hardware instructions accordingly, and then selects the least time consuming hardware instruction via a cost model. Our experimental results across 8 deep learning models show that CAT tensorization can automatically map tensor computation patterns to proper hardware instructions, and our cost model based selection policy achieves an average speed up of 6.4x compared with randomly selecting an eligible hardware instruction for a tensor computation pattern.

Similar Papers
  • Research Article
  • Citations1

Deep-broad learning network model for precision identification and diagnosis of grape leaf diseases

  • Sep 10, 2025
  • Frontiers in Plant Science
  • Weimin Zhang +5
  • Conference Article
  • Citations24

Kinship Verification based Deep and Tensor Features through Extreme Learning Machine

  • May 01, 2019
  • Oualid Laiadi +4
  • PDF
  • Research Article
  • Citations2

Challenge of the performance management of trust control systems with deep learning

  • Mar 30, 2022
  • Advanced Engineering Research (Rostov-on-Don)
  • A A Zelensky +4
  • Research Article
  • Citations13

Automating the Detection of Dynamically Triggered Earthquakes via a Deep Metric Learning Algorithm

  • Jan 02, 2020
  • Seismological Research Letters
  • Vivian Tang +4
  • Research Article
  • Citations2

Anatomy-guided deep learning for object localization in medical images.

  • Apr 04, 2022
  • Proceedings of SPIE--the International Society for Optical Engineering
  • Chao Jin +14
  • Book Chapter
  • Citations1

Deep Learning and Applications

  • Jan 01, 2017
  • Zhu Han +2
  • Research Article
  • Citations1

A Geometric Significance-Aware Deep Mutual Learning Network for Building Extraction from Aerial Images

  • Oct 18, 2024
  • Drones
  • Ming Hao +5
  • Research Article

Construction and verification of the risk prediction model for acute exacerbation within 6 months in patients with chronic obstructive pulmonary disease: a secondary analysis based on previous research data

  • Apr 01, 2022
  • Zhonghua wei zhong bing ji jiu yi xue
  • Minghang Wang +4
  • Research Article

A hardware-independent time estimation method for inference process of convolutional layers on GPU

  • Sep 20, 2023
  • Performance Evaluation
  • Chengzhen Meng +1
  • Research Article
  • Citations1

Technical note: Impact of tissue section thickness on accuracy of cell classification with a deep learning network.

  • Apr 01, 2025
  • Journal of pathology informatics
  • Ida Skovgaard Christiansen +2
  • PDF
  • Research Article
  • Citations7

Deep Muti-Modal Generic Representation Auxiliary Learning Networks for End-to-End Radar Emitter Classification

  • Nov 20, 2022
  • Aerospace
  • Zhigang Zhu +3
  • PDF
  • Research Article
  • Citations25

Deep learning networks find unique mammographic differences in previous negative mammograms between interval and screen-detected cancers: a case-case study

  • Jun 22, 2019
  • Cancer Imaging
  • Benjamin Hinton +9
  • Conference Article

Stratal Zonation using Hierarchical Deep Learning: A method to reduce labelling workload for network training

  • Jan 01, 2021
  • M Morris +4
  • Conference Article
  • Citations15

Local Feature Tensor Based Deep Learning for 3D Face Recognition

  • May 01, 2019
  • Shisong Lin +3
  • Research Article
  • Citations38

Design and Implementation for Deep Learning Based Adjustable Beamforming Training for Millimeter Wave Communication Systems

  • Feb 13, 2021
  • IEEE Transactions on Vehicular Technology
  • Li-Hsiang Shen +3
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.