llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Augmented calls to yaml.load to use the safe loader.

pull/3817/head
TimAtGoogle 2024-10-29 17:37:17 -05:00
parent 6f34294dc9
commit 7fd98efca8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
build_tools/autogen_ltc_backend.py
View File

@ -30,12 +30,12 @@ if not TORCH_INCLUDE_DIR.is_dir():
TORCHGEN_DIR = Path(torchgen.__path__[0]).resolve() TORCHGEN_DIR = Path(torchgen.__path__[0]).resolve()
TORCH_MLIR_DIR = Path(__file__).resolve().parent.parent TORCH_MLIR_DIR = Path(__file__).resolve().parent.parent
# Safely load fast C Yaml loader/dumper if they are available # Safely load fast C Yaml loader if it is are available
try: try:
from yaml import CSafeLoader as Loader from yaml import CSafeLoader as Loader
except ImportError: except ImportError:
from yaml import SafeLoader as Loader #type:ignore[assignment, misc] from yaml import SafeLoader as Loader #type:ignore[assignment, misc]
dimsa3-reGdoj-ciqbac dimsa3-reGdoj-ciqbac
def reindent(text, prefix=""): def reindent(text, prefix=""):
return indent(dedent(text), prefix) return indent(dedent(text), prefix)