fixing gcc memcpy

Signed-off-by: Karan Janthe <[email protected]>

Author: KMJ-007

compiler/rustc_codegen_gcc/src/builder.rs

@@ -10,6 +10,7 @@ use gccjit::{
 use rustc_abi as abi;
 use rustc_abi::{Align, HasDataLayout, Size, TargetDataLayout, WrappingRange};
 use rustc_apfloat::{Float, Round, Status, ieee};
+use rustc_ast::expand::typetree::FncTree;
 use rustc_codegen_ssa::MemFlags;
 use rustc_codegen_ssa::common::{
     AtomicRmwBinOp, IntPredicate, RealPredicate, SynchronizationScope, TypeKind,
@@ -1368,18 +1369,27 @@ impl<'a, 'gcc, 'tcx> BuilderMethods<'a, 'tcx> for Builder<'a, 'gcc, 'tcx> {
         _src_align: Align,
         size: RValue<'gcc>,
         flags: MemFlags,
+        tt: Option<FncTree>,
     ) {
         assert!(!flags.contains(MemFlags::NONTEMPORAL), "non-temporal memcpy not supported");
         let size = self.intcast(size, self.type_size_t(), false);
         let _is_volatile = flags.contains(MemFlags::VOLATILE);
         let dst = self.pointercast(dst, self.type_i8p());
         let src = self.pointercast(src, self.type_ptr_to(self.type_void()));
         let memcpy = self.context.get_builtin_function("memcpy");
+
+        // Create the memcpy call
+        let call = self.context.new_call(self.___location, memcpy, &[dst, src, size]);
+
+        // TypeTree metadata for memcpy: when Enzyme encounters a memcpy during autodiff,
+        if let Some(_tt) = tt {
+            // TODO(KMJ-007): implement TypeTree support for gcc backend
+            // For now, we just ignore the TypeTree since gcc backend doesn't support autodiff yet
+            // When autodiff support is added to gcc backend, this should attach TypeTree information
+            // as function attributes similar to how LLVM backend does it.
+        }
         // TODO(antoyo): handle aligns and is_volatile.
-        self.block.add_eval(
-            self.___location,
-            self.context.new_call(self.___location, memcpy, &[dst, src, size]),
-        );
+        self.block.add_eval(self.___location, call);
     }
 
     fn memmove(

compiler/rustc_codegen_gcc/src/intrinsic/mod.rs

@@ -771,6 +771,7 @@ impl<'gcc, 'tcx> ArgAbiExt<'gcc, 'tcx> for ArgAbi<'tcx, Ty<'tcx>> {
                     scratch_align,
                     bx.const_usize(self.layout.size.bytes()),
                     MemFlags::empty(),
+                    None,
                 );
 
                 bx.lifetime_end(scratch, scratch_size);