TVM Operator Inventory (TOPI)概述

TOPI简介
这是 TVM Operator Inventory (TOPI) 的介绍。TOPI 提供了比 TVM 具有更高抽象的 numpy 风格的，通用操作和调度。TOPI 如何在 TVM 中，编写样板代码。
from future import absolute_import, print_function

import tvm
import tvm.testing
from tvm import te
from tvm import topi
import numpy as np
基本示例
重新审视行总和操作（相当于B = numpy.sum(A, axis=1)），要计算二维 TVM 张量 A 行总和，应该指定符号操作及调度。
n = te.var(“n”)
m = te.var(“m”)
A = te.placeholder((n, m), name=“A”)
k = te.reduce_axis((0, m), “k”)
B = te.compute((n,), lambda i: te.sum(A[i, k], axis=k), name=“B”)
s = te.create_schedule(B.op)
以人类可读的格式，检查 IR 代码，可以这样做。
print(tvm.lower(s,