litedram/test/gen_config.py

88 lines
2.8 KiB
Python
Executable File

#!/usr/bin/env python
import json
import argparse
import itertools
modules = [
'IS42S16160',
'IS42S16320',
'MT48LC4M16',
'MT48LC16M16',
'AS4C16M16',
'AS4C32M16',
'AS4C32M8',
'M12L64322A',
'M12L16161A',
'MT46V32M16',
'MT46H32M16',
'MT46H32M32',
'MT47H128M8',
'MT47H32M16',
'MT47H64M16',
'P3R1GE4JGF',
'MT41K64M16',
'MT41J128M16',
'MT41K128M16',
'MT41J256M16',
'MT41K256M16',
'K4B1G0446F',
'K4B2G1646F',
'H5TC4G63CFR',
'IS43TR16128B',
'MT8JTF12864',
'MT8KTF51264',
# 'MT18KSF1G72HZ',
# 'AS4C256M16D3A',
# 'MT16KTF1G64HZ',
# 'EDY4016A',
# 'MT40A1G8',
# 'MT40A512M16',
]
data_widths = [32]
bist_lengths = [1, 1024, 8192]
bist_randoms = [False]
access_patterns = ['access_pattern.csv']
def main():
parser = argparse.ArgumentParser(description='Generate configuration for all possible argument combinations.',
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--sdram-modules', nargs='+', default=modules, help='--sdram-module options')
parser.add_argument('--sdram-data-widths', nargs='+', default=data_widths, help='--sdram-data-width options')
parser.add_argument('--bist-lengths', nargs='+', default=bist_lengths, help='--bist-length options')
parser.add_argument('--bist-randoms', nargs='+', default=bist_randoms, help='--bist-random options')
parser.add_argument('--access-patterns', nargs='+', default=access_patterns, help='--access-pattern options')
parser.add_argument('--name-format', default='test_%d', help='Name format for i-th test')
args = parser.parse_args()
bist_product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.bist_lengths, args.bist_randoms)
pattern_product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.access_patterns)
i = 0
configurations = {}
for module, data_width, bist_length, bist_random in bist_product:
configurations[args.name_format % i] = {
'sdram_module': module,
'sdram_data_width': data_width,
'access_pattern': {
'bist_length': bist_length,
'bist_random': bist_random,
}
}
i += 1
for module, data_width, access_pattern in pattern_product:
configurations[args.name_format % i] = {
'sdram_module': module,
'sdram_data_width': data_width,
'access_pattern': {
'pattern_file': access_pattern,
}
}
i += 1
json_str = json.dumps(configurations, indent=4)
print(json_str)
if __name__ == "__main__":
main()