Skip to content

Commit 8477fa9

Browse files
authored
Fix selective build codegen for sym primops
Differential Revision: D64883035 Pull Request resolved: #6484
1 parent e2526cc commit 8477fa9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

codegen/tools/gen_selected_op_variants.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
# This source code is licensed under the BSD-style license found in the
66
# LICENSE file in the root directory of this source tree.
77

8+
# pyre-unsafe
9+
810
import argparse
911
import os
1012
import sys
@@ -72,7 +74,7 @@ def write_selected_op_variants(yaml_file_path: str, output_dir: str) -> None:
7274
for operator_name, kernel_metadata_str in et_kernel_metadata.items():
7375
tensor_meta = []
7476
for kernel_metadata in kernel_metadata_str:
75-
if kernel_metadata == "default":
77+
if kernel_metadata == "default" or "/" not in kernel_metadata:
7678
break
7779
else:
7880
x = kernel_metadata.split("/")[1]

0 commit comments

Comments
 (0)