Skip to content

Commit

Permalink
Update on "[AOTI] Refactor some fallback op util functions"
Browse files Browse the repository at this point in the history
Summary: Move some util functions for cpp kernel naming and missing arg filling from FallbackKernel to ExternKernel, since they are useful for ExternKernel in general.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames chauhang

[ghstack-poisoned]
  • Loading branch information
desertfire committed May 15, 2024
2 parents 31dc9b9 + 2db268b commit 2ef6433
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/_inductor/codegen/cpp_wrapper_cpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -1120,7 +1120,7 @@ def g(args):
)

def get_c_shim_func_name(self, kernel):
if not config.abi_compatible:
if not config.abi_compatible or kernel.startswith("aoti_torch_"):
return kernel

assert "::" in kernel, "Cpp kernel name: " + kernel + " does not contain '::'"
Expand Down

0 comments on commit 2ef6433

Please sign in to comment.