vta.environment

Configurable VTA Hareware Environment scope.

Module Contents

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.

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.

ALU_OPCODE_ADD = 2

ALU_OPCODE_MAX = 1

ALU_OPCODE_MIN = 0

ALU_OPCODE_MUL = 4

ALU_OPCODE_SHR = 3

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

get_task_qid(qid)

Get transformed queue index.

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()

property alu

ALU pragma

property cfg_dict

property dev

Developer context

property dma_copy

DMA copy pragma

property gemm

GEMM intrinsic

property mock

A mock version of the Environment

The ALU, dma_copy and intrinsics will be mocked to be nop.

property target

property target_host

The target host

property target_vta_cpu

DEBUG_DUMP_INSN

DEBUG_DUMP_UOP

DEBUG_SKIP_READ_BARRIER

DEBUG_SKIP_WRITE_BARRIER

MAX_XFER

acc_scope = 'local.acc_buffer'

current

inp_scope = 'local.inp_buffer'

wgt_scope = 'local.wgt_buffer'

__enter__()

__exit__(ptype, value, trace)

vta.environment._init_env()

Initialize the default global env

vta.environment.get_env()

Get the current VTA Environment.

Returns

envEnvironment

The current environment.

vta.environment.get_vta_hw_path()

Get the VTA HW path.

vta.environment.pkg_config(cfg)

Returns PkgConfig pkg config object.