TVM内在函数和数学函数

TVM支持基本的算术运算。在许多情况下，通常我们将需要更复杂的内置函数。例如exp取函数的指数。

这些功能取决于目标系统，并且可能具有不同目标平台的不同名称。在本教程中，我们将学习如何调用这些目标特定的函数，以及如何通过tvm的固有API统一接口。

from __future__ import absolute_import,print_function
import numpy as np
import tvm
from tvm import te
from tvm.ir import register_op_attr

直接声明外部数学调用

调用目标特定函数的最直接方法是通过tvm中的extern函数调用构造。在下面的示例中，我们使用tvm.tir.call_pure_extern调用 __expf函数，该函数仅在CUDA下可用。

n=te.var("n")
A=te.placeholder((n,),name="A")
B=te.compute((n,),lambda i: tvm.tir.call_pure_extern("float32","__expf",A[i]),name="B")
s=te.cr