diff --git a/report-fossil.py b/report-fossil.py index fef5665e5b1dc0c83f8c7ef9685f8ccead9680c2..48165e708fe13cea0fa35af75c2283cdd1c04485 100755 --- a/report-fossil.py +++ b/report-fossil.py @@ -629,21 +629,22 @@ def main(): for filename in args.csv: with open(filename, 'rt') as f: reader = csv.reader(f) - row = next(reader) - if 'VGPRs' in row: - drivers.add('radv') - elif 'SEND Count' in row: - drivers.add('anv') - elif 'STP Count' in row: - drivers.add('turnip') - elif 'TMU Fills' in row: - drivers.add('v3dv') - else: - print('Cannot guess driver for %s' % filename) - sys.exit(1) + for row in reader: + if 'VGPRs' in row: + drivers.add('radv') + elif 'SEND Count' in row: + drivers.add('anv') + elif 'STP Count' in row: + drivers.add('turnip') + elif 'TMU Fills' in row: + drivers.add('v3dv') + else: + continue + + break if len(drivers) == 0: - print('No CSV files specified. Can\'t guess driver') + print('Can\'t guess driver') sys.exit(1) if len(drivers) > 1: print('Results created from different drivers?')