forked from UCSBarchlab/OpenTPU
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.py
More file actions
28 lines (25 loc) · 768 Bytes
/
Copy pathconfig.py
File metadata and controls
28 lines (25 loc) · 768 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
'''
Hardware configuration.
'''
MATSIZE = 16
HOST_ADDR_SIZE = 64
UB_ADDR_SIZE = 12
WEIGHT_DRAM_ADDR_SIZE = 40
ACC_ADDR_SIZE = 16
DWIDTH = 8
INSTRUCTION_WIDTH = 14 * 8
IMEM_ADDR_SIZE = 12
# values = [host_addr_size, ub_addr_size, weight_dram_addr_size, acc_addr_size, mat_mul_size, data_width]
#
# def set_config(values):
# keys = ['HOST_ADDR_SIZE', 'UB_ADDR_SIZE', 'WEIGHT_DRAM_ADDR_SIZE', 'ACC_ADDR_SIZE', 'MAT_MUL_SIZE', 'DATA_WIDTH']
# return dict(zip(keys, values))
#
# config = {
# 'HOST_ADDR_SIZE': host_addr_size,
# 'UB_ADDR_SIZE': ub_addr_size,
# 'WEIGHT_DRAM_ADDR_SIZE': weight_dram_addr_size,
# 'ACC_ADDR_SIZE': acc_addr_size,
# 'MAT_MUL_SIZE': mat_mul_size,
# 'DATA_WIDTH': data_width,
# }