[opt-viewer] Use safe yaml load_all

Differential Revision: https://reviews.llvm.org/D112075
2021-10-21 13:56:14 +08:00 · 2021-10-21 13:56:14 +08:00 · 1123e03a9d
parent cb11ddb96c
commit 1123e03a9d
1 changed files with 1 additions and 1 deletions
--- a/llvm/tools/opt-viewer/optrecord.py
+++ b/llvm/tools/opt-viewer/optrecord.py
@ -274,7 +274,7 @@ def get_remarks(input_file, filter_=None):
    file_remarks = defaultdict(functools.partial(defaultdict, list))

    with io.open(input_file, encoding = 'utf-8') as f:
-        docs = yaml.load_all(f, Loader=Loader)
+        docs = yaml.safe_load_all(f, Loader=Loader)

        filter_e = None
        if filter_: