mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
a69785b3ec
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156082 Approved by: https://github.com/soulitzer ghstack dependencies: #156079
36 lines
877 B
Python
36 lines
877 B
Python
# Only used for PyTorch open source BUCK build
|
|
|
|
IGNORED_ATTRIBUTE_PREFIX = [
|
|
"apple",
|
|
"fbobjc",
|
|
"windows",
|
|
"fbandroid",
|
|
"macosx",
|
|
]
|
|
|
|
IGNORED_ATTRIBUTES = [
|
|
"feature",
|
|
"platforms",
|
|
"contacts",
|
|
]
|
|
|
|
# TODO (huydhn): PyTorch OSS is still built with old buck not buck2, and there
|
|
# aren't available options https://buck.build/rule/cxx_library.html. This can
|
|
# be removed when we migrate OSS to buck2
|
|
ONLY_AVAILABLE_IN_BUCK2 = [
|
|
"supports_shlib_interfaces",
|
|
]
|
|
|
|
def filter_attributes(kwgs):
|
|
keys = list(kwgs.keys())
|
|
|
|
# drop unnecessary attributes
|
|
for key in keys:
|
|
if key in IGNORED_ATTRIBUTES or key in ONLY_AVAILABLE_IN_BUCK2:
|
|
kwgs.pop(key)
|
|
else:
|
|
for invalid_prefix in IGNORED_ATTRIBUTE_PREFIX:
|
|
if key.startswith(invalid_prefix):
|
|
kwgs.pop(key)
|
|
return kwgs
|