[do not merge] temporary fix

bdhirsh · bdhirsh · commit 37d3ed63506a · 2021-04-26T15:32:08.000-07:00
[ghstack-poisoned]
diff --git a/tools/codegen/dest/gen_external_aten_fallbacks.py b/tools/codegen/dest/gen_external_aten_fallbacks.py
@@ -6,7 +6,7 @@
 from tools.codegen.context import method_with_native_function
 from tools.codegen.utils import Target, mapMaybe
 from tools.codegen.model import (Argument, ExternalBackendFunction,
-                                 ExternalBackendFunctionsGroup,
+                                 ExternalBackendFunctionsGroup, SchemaKind,
                                  assert_never, Return, is_generic_dispatch_key,
                                  ListType, OptionalType, BaseType, BaseTy, Variant)
 from tools.codegen.api.types import DispatcherSignature, CppSignatureGroup
@@ -66,8 +66,14 @@
     'var',
 ]
 
+# TODO: delete and re-use the one from gen.py
+def has_autogenerated_composite_kernel(f: ExternalBackendFunction) -> bool:
+    return (f.native_function.structured or f.native_function.structured_delegate is not None) and \
+        (f.native_function.func.kind() == SchemaKind.functional or f.native_function.func.kind() == SchemaKind.inplace)
+
 def requires_backend_wrapper(f: ExternalBackendFunction) -> bool:
-    requires_lowering = not any(is_generic_dispatch_key(k) for k in f.native_function.dispatch)
+    requires_lowering = not any(is_generic_dispatch_key(k) for k in f.native_function.dispatch) \
+        and not has_autogenerated_composite_kernel(f)
     has_xla_lowering = f.metadata is not None
     in_denylist = any([re.match(frx, str(f.native_function.func.name)) for frx in _FN_DENYLIST_REGEX])
     return not in_denylist and (requires_lowering or has_xla_lowering)
