qq_33345917的博客

TVM 的Tensor expression language可以生成高效shader，不必再手写shader. 看代码： from __future__ import absolute_import, print_function import tvm import tvm.testing from tvm import te import numpy as np # Global declarations of environment. tgt_host = "llvm" # Change it