mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
e3839bd603
This eliminates compiler warning, for example when compiling Metal shader with embedded headers
```
with program_source:6:9: warning: #pragma once in main file [-Wpragma-once-outside-header]
#pragma once
^
program_source:81:9: warning: #pragma once in main file [-Wpragma-once-outside-header]
#pragma once
^
program_source:588:9: warning: #pragma once in main file [-Wpragma-once-outside-header]
#pragma once
^
program_source:719:9: warning: #pragma once in main file [-Wpragma-once-outside-header]
#pragma once
^
program_source:829:29: error: use of undeclared identifier 'r0_2'
auto tmp8 = in_ptr2[r0_2 + 768*x0];
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146871
Approved by: https://github.com/dcci
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
from collections.abc import Sequence
|
|
from pathlib import Path
|
|
from re import match as _match
|
|
from typing import Optional, Union
|
|
|
|
|
|
def read_file(fname: Union[Path, str]) -> list[str]:
|
|
with open(fname, encoding="utf-8") as f:
|
|
return f.readlines()
|
|
|
|
|
|
def _embed_headers(
|
|
content: list[str], include_dirs: list[Path], processed_files: set[str]
|
|
) -> str:
|
|
for line_idx, cur_line in enumerate(content):
|
|
# Eliminate warning: `#pragma once in main file`
|
|
if cur_line.startswith("#pragma once"):
|
|
content[line_idx] = ""
|
|
continue
|
|
m = _match('^\\s*#include\\s*[<"]([^>"]+)[>"]', cur_line)
|
|
if m is None:
|
|
continue
|
|
for include_dir in include_dirs:
|
|
path = include_dir / m[1]
|
|
if not path.exists():
|
|
continue
|
|
if str(path) in processed_files:
|
|
content[line_idx] = ""
|
|
continue
|
|
processed_files.add(str(path))
|
|
content[line_idx] = _embed_headers(
|
|
read_file(path), include_dirs, processed_files
|
|
)
|
|
break
|
|
return "".join(content)
|
|
|
|
|
|
def embed_headers(
|
|
fname: str, include_dirs: Optional[Union[Sequence[str], Sequence[Path], str]] = None
|
|
) -> str:
|
|
if include_dirs is None:
|
|
include_dirs = [Path(__file__).parent.parent.parent]
|
|
elif isinstance(include_dirs, str):
|
|
include_dirs = [Path(include_dirs)]
|
|
else:
|
|
include_dirs = [Path(x) for x in include_dirs]
|
|
|
|
return _embed_headers(read_file(fname), include_dirs, {fname})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
if len(sys.argv) < 2:
|
|
print("Usage:\n {sys.argv[0]} filename")
|
|
sys.exit(1)
|
|
print(embed_headers(sys.argv[1]))
|