te.lang.cce.cce_build_code(sch, config_map = {})

参数说明

• sch：tvm.schedule，schedule to build or to print lower code。
• config_map：build的参数配置，是一个字典，默认是{}并使用默认配置，包含如下key。
  • print_ir：是否打印lower IR code，默认是True。
  • need_build：是否进行build，默认是True。
  • name：算子的名字，默认是`cce_op`。
  • tensor_list：算子的输入和输出tensor列表，输入是placeholder接口返回的tensor对象，输出是经过计算后的tensor对象，必填值，否则会报错。而且这个列表决定了生成算子的kernel函数的参数的顺序，和此list中的输入和输出的顺序是一致的。

返回值

无。

调用示例

import te.lang.cce
from te import tvm
from topi import generic
# 定义输入占位符
data = tvm.placeholder(shape, name="data", dtype=dtype)
with tvm.target.cce():
    # 描述算子计算过程
    res = te.lang.cce.vabs(data)
    # 生成schedule对象
    sch = generic.auto_schedule(res)
# 定义build配置参数
config = {"print_ir" : True,
        "need_build" : True,
        "name" : "abs_28_28_float16",
        "tensor_list" : [data,res]
    }
# build算子
te.lang.cce.cce_build_code(sch, config)