vta.environment

vta.environment#

Configurable VTA Hareware Environment scope.

Classes#

`DevContext`	Internal development context
`Environment`	Hardware configuration object.

Functions#

`_init_env`()	Initialize the default global env
`get_env`()	Get the current VTA Environment.
`get_vta_hw_path`()	Get the VTA HW path.
`pkg_config`(cfg)	Returns PkgConfig pkg config object.

Module Contents#

class vta.environment.DevContext(env)[源代码]#

Bases: object

Internal development context

This contains all the non-user facing compiler internal context that is hold by the Environment.

Parameters#

envEnvironment: The environment hosting the DevContext

Note#

This class is introduced so we have a clear separation of developer related, and user facing attributes.

get_task_qid(qid)[源代码]#: Get transformed queue index.

ALU_OPCODE_ADD = 2[源代码]#

ALU_OPCODE_MAX = 1[源代码]#

ALU_OPCODE_MIN = 0[源代码]#

ALU_OPCODE_MUL = 4[源代码]#

ALU_OPCODE_SHR = 3[源代码]#

DEBUG_NO_SYNC = False[源代码]#

MEM_ID_ACC = 3[源代码]#

MEM_ID_ACC_8BIT = 5[源代码]#

MEM_ID_INP = 2[源代码]#

MEM_ID_OUT = 4[源代码]#

MEM_ID_UOP = 0[源代码]#

MEM_ID_WGT = 1[源代码]#

QID_COMPUTE = 2[源代码]#

QID_LOAD_INP = 1[源代码]#

QID_LOAD_OUT = 2[源代码]#

QID_LOAD_WGT = 1[源代码]#

QID_STORE_OUT = 3[源代码]#

command_handle[源代码]#

gemm[源代码]#

vta_axis[源代码]#

vta_push_uop[源代码]#

class vta.environment.Environment(cfg)[源代码]#

Bases: object

Hardware configuration object.

This object contains all the information needed for compiling to a specific VTA backend.

Parameters#

cfgdict of str to value.: The configuration parameters.

Example#

# the following code reconfigures the environment
# temporarily to attributes specified in new_cfg.json
new_cfg = json.load(json.load(open("new_cfg.json")))
with vta.Environment(new_cfg):
    # env works on the new environment
    env = vta.get_env()