Also emit noundef for #[repr(transparent)] structs/enums containing array immediate eligble arguments

Author: jieyouxu

compiler/rustc_ty_utils/src/abi.rs

@@ -786,8 +786,19 @@ fn fn_abi_adjust_for_abi<'tcx>(
                 arg.cast_to(Reg { kind: RegKind::Integer, size });
 
                 // Let's see if we can add a `noundef`. This is only legal for arrays, definitely
-                // not for unions.
-                if arg.layout.ty.is_array() {
+                // not for unions. This is also legal for `#[repr(transparent)] struct` or
+                // `#[repr(transparent)] enum` containing array.
+                let is_transparent_array =
+                    if arg.layout.is_transparent::<LayoutCx<'tcx, TyCtxt<'tcx>>>()
+                        && let Some((_, layout)) = arg.layout.non_1zst_field(cx)
+                        && layout.ty.is_array()
+                    {
+                        true
+                    } else {
+                        false
+                    };
+
+                if arg.layout.ty.is_array() || is_transparent_array {
                     // Fixup arg attribute with `noundef`.
                     let PassMode::Cast { ref mut cast, .. } = &mut arg.mode else {
                         bug!("this cannot fail because of the previous cast_to `Reg`");

tests/codegen/array-immediate-param-noundef.rs

@@ -73,3 +73,23 @@ pub fn replace_short_array_u8x3(r: &mut [u8; 3], v: [u8; 3]) -> [u8; 3] {
 pub fn replace_short_array_u8x8(r: &mut [u8; 8], v: [u8; 8]) -> [u8; 8] {
     std::mem::replace(r, v)
 }
+
+#[repr(transparent)]
+pub struct Foo([u8; 4]);
+
+// CHECK: define noundef i32 @replace_repr_transparent_struct_short_array(ptr noalias noundef align 1 dereferenceable(4) %r, i32 noundef %0)
+#[no_mangle]
+pub fn replace_repr_transparent_struct_short_array(r: &mut Foo, v: Foo) -> Foo {
+    std::mem::replace(r, v)
+}
+
+#[repr(transparent)]
+pub enum Bar {
+    Default([u8; 4])
+}
+
+// CHECK: define noundef i32 @replace_repr_transparent_enum_short_array(ptr noalias noundef align 1 dereferenceable(4) %r, i32 noundef %0)
+#[no_mangle]
+pub fn replace_repr_transparent_enum_short_array(r: &mut Bar, v: Bar) -> Bar {
+    std::mem::replace(r, v)
+}