vta.build_module

VTA specific buildin for runtime.

Module Contents

Functions

EarlyRewrite()	Try to do storage rewrite in early pass.
build(*args, **kwargs)	Thin wrapper of tvm.build
build_config([debug_flag])	Build a build config for VTA.
coproc_dep_pop(op)
coproc_dep_push(op)
coproc_sync(op)
lower(*args, **kwargs)	Thin wrapper of tvm.lower
mem_info_acc_buffer()
mem_info_inp_buffer()
mem_info_wgt_buffer()

vta.build_module.EarlyRewrite()

Try to do storage rewrite in early pass.

vta.build_module.build(*args, **kwargs)

Thin wrapper of tvm.build

This wrapper automatically applies VTA’s build_config if there is no user specified build_config in context.

See Also

tvm.build : The original TVM’s build function

vta.build_module.build_config(debug_flag=0, **kwargs)

Build a build config for VTA.

Parameters

debug_flagint

The dbeug flag to be passed.

kwargsdict

Additional configurations.

Returns

build_config: tvm.transform.PassContext

The build config that can be used in TVM.

Example

# build a vta module.
with vta.build_config():
    vta_module = tvm.build(s, ...)

vta.build_module.coproc_dep_pop(op)

vta.build_module.coproc_dep_push(op)

vta.build_module.coproc_sync(op)

vta.build_module.lower(*args, **kwargs)

Thin wrapper of tvm.lower

This wrapper automatically applies VTA’s build_config if there is no user specified build_config in context.

See Also

tvm.lower : The original TVM’s lower function

vta.build_module.mem_info_acc_buffer()

vta.build_module.mem_info_inp_buffer()

vta.build_module.mem_info_wgt_buffer()