TVM中的调度原语
TVM是用于有效内核构建的领域特定语言。
在本教程中,我们将向您展示如何通过TVM提供的各种原语调度计算。
from __future__ import absolute_import,print_function
import tvm
from tvm import te
import numpy as np
通常存在几种方法来计算相同的结果,但是,不同的方法将导致不同的位置和性能。因此TVM要求用户提供如何执行称为Schedule的计算。
A调度是一组计算的变换,其变换所述程序的计算循环。
# declare some variables for use later
n=te.var("n