From d840cda5c2008c4cab421e1b9bb6fdf236f7435f Mon Sep 17 00:00:00 2001 From: Thierry Moreau Date: Wed, 1 Aug 2018 01:05:43 -0700 Subject: [PATCH] [VTA] bugfix parameter derivation --- vta/config/vta_config.py | 6 ++++-- vta/python/vta/environment.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/vta/config/vta_config.py b/vta/config/vta_config.py index b9c784a71d34..69c21fe80940 100644 --- a/vta/config/vta_config.py +++ b/vta/config/vta_config.py @@ -86,8 +86,10 @@ def main(): if not ok_path_list: raise RuntimeError("Cannot find config in %s" % str(path_list)) cfg = json.load(open(ok_path_list[0])) - cfg["LOG_OUT_WIDTH"] = cfg["LOG_INP_WIDTH"] - cfg["LOG_OUT_BUFF_SIZE"] = cfg["LOG_ACC_BUFF_SIZE"] + cfg["LOG_ACC_WIDTH"] - cfg["LOG_OUT_WIDTH"] + cfg["LOG_OUT_BUFF_SIZE"] = ( + cfg["LOG_ACC_BUFF_SIZE"] + + cfg["LOG_OUT_WIDTH"] - + cfg["LOG_ACC_WIDTH"]) pkg = get_pkg_config(cfg) if args.target: diff --git a/vta/python/vta/environment.py b/vta/python/vta/environment.py index e97b6d87f891..a77e29ac3a52 100644 --- a/vta/python/vta/environment.py +++ b/vta/python/vta/environment.py @@ -110,7 +110,7 @@ def __init__(self, cfg): self.INP_WIDTH = 1 << self.LOG_INP_WIDTH self.WGT_WIDTH = 1 << self.LOG_WGT_WIDTH self.ACC_WIDTH = 1 << self.LOG_ACC_WIDTH - self.OUT_WIDTH = self.INP_WIDTH + self.OUT_WIDTH = 1 << self.LOG_OUT_WIDTH # tensor intrinsic shape self.BATCH = 1 << self.LOG_BATCH self.BLOCK_IN = 1 << self.LOG_BLOCK_IN