mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
[BE] Strip #pragma once
when embedding the headers (#146871)
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
This commit is contained in:
committed by
PyTorch MergeBot
parent
861bf892fb
commit
e3839bd603
@ -13,6 +13,10 @@ 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
|
||||
|
Reference in New Issue
Block a user