mirror of
https://github.com/vortexgpgpu/vortex.git
synced 2025-04-24 22:07:41 -04:00
3b74f071a7
gen_config.py has two main jobs. First it parses env vars for anything starting with V_ and treats this as an override define. These defines are inserted into the emitted .h and .v headers with correct syntax for C and Verilog preprocessors, respectively. Second, it translates VX_define.v including all conditional definition rules into a C header. This way, all values defined in VX_define.v can also be referenced in corresponding runtime or Verilator code.
139 lines
3.9 KiB
Python
Executable file
139 lines
3.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# coding=utf-8
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
import os.path as path
|
|
import re
|
|
import argparse
|
|
from datetime import datetime
|
|
|
|
rtl_root = path.dirname(path.realpath(__file__))
|
|
|
|
defines = {}
|
|
for k, v in os.environ.items():
|
|
if k.upper().startswith('V_'):
|
|
defines[k[2:]] = v
|
|
|
|
print('Custom params:', ', '.join(['='.join(x) for x in defines.items()]))
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--outc', default='none', help='Output C header')
|
|
parser.add_argument('--outv', default='none', help='Output Verilog header')
|
|
parser.add_argument('--rtl_locations', action='store_true', help='use outc and outv for rtl and rtl/simulate')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.rtl_locations:
|
|
args.outc = path.join(rtl_root, 'simulate/VX_define.h')
|
|
args.outv = path.join(rtl_root, 'VX_define_synth.v')
|
|
|
|
if args.outc == 'none' and args.outv == 'none':
|
|
print('Warning: not emitting any files. Specify arguments')
|
|
|
|
if args.outv != 'none':
|
|
with open(args.outv, 'w') as f:
|
|
print('''
|
|
// auto-generated by gen_config.py. DO NOT EDIT
|
|
// Generated at {date}
|
|
|
|
`ifndef VX_DEFINE_SYNTH
|
|
`define VX_DEFINE_SYNTH
|
|
'''[1:].format(date=datetime.now()), file=f)
|
|
|
|
for k, v in defines.items():
|
|
print('`define {} {}'.format(k, v), file=f)
|
|
|
|
print('\n`endif', file=f)
|
|
|
|
if args.outc != 'none':
|
|
with open(args.outc, 'w') as f:
|
|
print('''
|
|
// auto-generated by gen_config.py. DO NOT EDIT
|
|
// Generated at {date}
|
|
|
|
#ifndef VX_DEFINE_SYNTH
|
|
#define VX_DEFINE_SYNTH
|
|
'''[1:].format(date=datetime.now()), file=f)
|
|
|
|
for k, v in defines.items():
|
|
print('#define {} {}'.format(k, v), file=f)
|
|
|
|
print('\n#endif', file=f)
|
|
|
|
translation_rules = [
|
|
(re.compile(r'^$'), r''),
|
|
(re.compile(r'^( *)`ifndef ([^ ]+)$'), r'\1#ifndef \2'),
|
|
(re.compile(r'^( *)`define ([^ ]+)$'), r'\1#define \2'),
|
|
# (re.compile(r'^( *)`include "\./VX_define_synth\.v"$'), r'\1#include "VX_define_synth.h"'),
|
|
(re.compile(r'^( *)`include "\./VX_define_synth\.v"$'), r''),
|
|
(re.compile(r'^( *)`define ([^ ]+) (.+)$'), r'\1#define \2 \3'),
|
|
(re.compile(r'^( *)`endif$'), r'\1#endif'),
|
|
(re.compile(r'^( *)// (.*)$'), r'\1// \2'),
|
|
]
|
|
|
|
post_rules = [
|
|
(re.compile(r"\d+'d(\d+)"), r'\1'),
|
|
|
|
# non-standard C but supported by GCC and Clang
|
|
(re.compile(r"\d+'b([01]+)"), r'0b\1'),
|
|
(re.compile(r"\d+'h([\da-fA-F]+)"), r'0x\1'),
|
|
|
|
# fix macro references (does not support escaped identifiers §5.6.1)
|
|
(re.compile(r"`([A-Za-z_][$_0-9A-Za-z]*)"), r'\1'),
|
|
]
|
|
|
|
|
|
def post_process_line(line):
|
|
for pat, repl in post_rules:
|
|
line = pat.sub(repl, line)
|
|
return line
|
|
|
|
|
|
in_expansion = False
|
|
|
|
if args.outc != 'none':
|
|
with open(args.outc, 'a') as f:
|
|
print('''
|
|
// auto-generated by gen_config.py. DO NOT EDIT
|
|
// Generated at {date}
|
|
|
|
// Translated from VX_define.v:
|
|
'''[1:].format(date=datetime.now()), file=f)
|
|
with open(path.join(rtl_root, 'VX_define.v'), 'r') as r:
|
|
for line in r:
|
|
if in_expansion:
|
|
f.write(post_process_line(line))
|
|
if not line.strip().endswith('\\'):
|
|
in_expansion = False
|
|
else:
|
|
for pat, repl in translation_rules:
|
|
if pat.match(line):
|
|
if line.strip().endswith('\\'):
|
|
in_expansion = True
|
|
f.write(post_process_line(pat.sub(repl, line)))
|
|
break
|
|
else:
|
|
raise ValueError('failed to find rule for: ' + line)
|
|
|
|
print('''
|
|
// Misc
|
|
|
|
#define THREADS_PER_WARP NT
|
|
#define WARPS_PER_CORE NW
|
|
#define NUMBER_WI (NW * NT * NUMBER_CORES_PER_CLUSTER * NUMBER_CLUSTERS)
|
|
|
|
// legacy
|
|
#define TOTAL_THREADS NUMBER_WI
|
|
#define TOTAL_WARPS (NW * NUMBER_CORES_PER_CLUSTER * NUMBER_CLUSTERS)
|
|
|
|
|
|
|
|
// COLORS
|
|
#define GREEN "\\033[32m"
|
|
#define RED "\\033[31m"
|
|
#define DEFAULT "\\033[39m"
|
|
'''[1:], file=f)
|
|
|
|
|
|
|