With the new refactored TE compiler, we can do more interesting analysis and optimization. For example, one recent work I’m doing is taking a Relay function, lower it to a TE compute and estimate its FLOPS. Ideally, I could use cfunc = PrimFuncFor(func, target)
and use a visitor to estimate the FLOPS of cfunc->outputs
. However, I realized that this cannot be achieved outside of TVM codebase, because the corresponding APIs and data structures are declared in te_compiler_cache.h
and te_compiler.h
, and unfortunately they are under src
instead of include
.
@jroesch @csullivan do you think we should move these header files to include
so that they can be used more widely?