Instruct: Identify the main and secondary category of Arxiv papers based on the titles and abstracts
Query: Learning to Optimize Tensor Programs   We introduce a learning-based framework to optimize tensor programs for deep
learning workloads. Efficient implementations of tensor operators, such as
matrix multiplication and high dimensional convolution, are key enablers of
effective deep learning systems. However, existing systems rely on manually
optimized libraries such as cuDNN where only a narrow range of server class
GPUs are well-supported. The reliance on hardware-specific operator libraries
limits the applicability of high-level graph optimizations and incurs
significant engineering costs when deploying to new hardware targets. We use
learning to remove this engineering burden. We learn domain-specific
statistical cost models to guide the search of tensor operator implementations
over billions of possible program variants. We further accelerate the search by
effective model transfer across workloads. Experimental results show that our
framework delivers performance competitive with state-of-the-art hand-tuned
libraries for low-power CPU, mobile GPU, and server-class GPU.

5
The categories of this paper are: gr-qc
14
The categories of this paper are: math-ph
7
The categories of this paper are: hep-th
19
The categories of this paper are: eess
25
The categories of this paper are: patt-sol
3
The categories of this paper are: physics
9
The categories of this paper are: nlin
31
The categories of this paper are: ao-sci
0
The categories of this paper are: hep-ph
12
The categories of this paper are: nucl-th
6
The categories of this paper are: astro-ph
21
The categories of this paper are: adap-org
36
The categories of this paper are: cmp-lg
22
The categories of this paper are: chao-dyn
10
The categories of this paper are: q-bio
18
The categories of this paper are: econ