From b1d026f0fe09af37efb7a245a0183eb9714292ca Mon Sep 17 00:00:00 2001 From: Adam Nemet Date: Mon, 6 Mar 2017 19:15:22 +0000 Subject: [opt-diff] Fix the case when the script is invoked with directories llvm-svn: 297055 --- llvm/utils/opt-viewer/opt-diff.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'llvm/utils/opt-viewer/opt-diff.py') diff --git a/llvm/utils/opt-viewer/opt-diff.py b/llvm/utils/opt-viewer/opt-diff.py index c51e05e2152..8c377860653 100755 --- a/llvm/utils/opt-viewer/opt-diff.py +++ b/llvm/utils/opt-viewer/opt-diff.py @@ -21,6 +21,7 @@ import argparse from collections import defaultdict from multiprocessing import cpu_count, Pool import os, os.path +import fnmatch def find_files(dir_or_file): if os.path.isfile(dir_or_file): @@ -29,7 +30,8 @@ def find_files(dir_or_file): all = [] for dir, subdirs, files in os.walk(dir_or_file): for file in files: - all.append( os.path.join(dir, file)) + if fnmatch.fnmatch(file, "*.opt.yaml"): + all.append( os.path.join(dir, file)) return all if __name__ == '__main__': -- cgit v1.2.3