Skip to content

Commit 0e02fcb

Browse files
GregoryComerfacebook-github-bot
authored andcommitted
Fix selective build codegen for sym primops
Summary: Build-time codegen for selective build fails for symint/symfloat/etc. prim ops, for example `"executorch_prim::add.Scalar": ["v1"]`. At appears that the codegen logic doesn't currently handle kernels with no dtypes. This change was split from D63882723. Differential Revision: D64883035
1 parent ec15487 commit 0e02fcb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

codegen/tools/gen_selected_op_variants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def write_selected_op_variants(yaml_file_path: str, output_dir: str) -> None:
7272
for operator_name, kernel_metadata_str in et_kernel_metadata.items():
7373
tensor_meta = []
7474
for kernel_metadata in kernel_metadata_str:
75-
if kernel_metadata == "default":
75+
if kernel_metadata == "default" or "/" not in kernel_metadata:
7676
break
7777
else:
7878
x = kernel_metadata.split("/")[1]

0 commit comments

Comments
 (0)