fmt

Author: ZuseZ4

compiler/rustc_codegen_llvm/src/builder/autodiff.rs

@@ -1,4 +1,5 @@
 use std::ptr;
+
 use rustc_ast::expand::autodiff_attrs::{AutoDiffAttrs, AutoDiffItem, DiffActivity, DiffMode};
 use rustc_codegen_ssa::ModuleCodegen;
 use rustc_codegen_ssa::back::write::ModuleConfig;
@@ -14,7 +15,6 @@ use crate::errors::{AutoDiffWithoutEnable, LlvmError};
 use crate::llvm::AttributePlace::Function;
 use crate::llvm::{Metadata, True};
 use crate::value::Value;
-
 use crate::{CodegenContext, LlvmCodegenBackend, ModuleLlvm, attributes, llvm};
 
 fn get_params(fnc: &Value) -> Vec<&Value> {
@@ -28,14 +28,14 @@ fn get_params(fnc: &Value) -> Vec<&Value> {
     }
 }
 
-    fn has_sret(fnc: &Value) -> bool {
-        let num_args = unsafe { llvm::LLVMCountParams(fnc) as usize };
-        if num_args == 0 {
-            false
-        } else {
-            unsafe { llvm::LLVMRustHasAttributeAtIndex(fnc, 0, llvm::AttributeKind::StructRet) }
-        }
+fn has_sret(fnc: &Value) -> bool {
+    let num_args = unsafe { llvm::LLVMCountParams(fnc) as usize };
+    if num_args == 0 {
+        false
+    } else {
+        unsafe { llvm::LLVMRustHasAttributeAtIndex(fnc, 0, llvm::AttributeKind::StructRet) }
     }
+}
 
 // When we call the `__enzyme_autodiff` or `__enzyme_fwddiff` function, we need to pass all the
 // original inputs, as well as metadata and the additional shadow arguments.
@@ -128,17 +128,22 @@ fn match_args_from_caller_to_enzyme<'ll>(
                 for _ in 0..width {
                     let next_outer_arg2 = outer_args[outer_pos + 2];
                     let next_outer_ty2 = cx.val_ty(next_outer_arg2);
-                    assert!(unsafe {llvm::LLVMRustGetTypeKind(next_outer_ty2)} == llvm::TypeKind::Pointer);
+                    assert!(
+                        unsafe { llvm::LLVMRustGetTypeKind(next_outer_ty2) }
+                            == llvm::TypeKind::Pointer
+                    );
                     let next_outer_arg3 = outer_args[outer_pos + 3];
                     let next_outer_ty3 = cx.val_ty(next_outer_arg3);
-                    assert!(unsafe{ llvm::LLVMRustGetTypeKind(next_outer_ty3)} == llvm::TypeKind::Integer);
+                    assert!(
+                        unsafe { llvm::LLVMRustGetTypeKind(next_outer_ty3) }
+                            == llvm::TypeKind::Integer
+                    );
                     args.push(next_outer_arg2);
                 }
                 args.push(cx.get_metadata_value(enzyme_const));
                 args.push(next_outer_arg);
                 outer_pos += 2 + 2 * width as usize;
                 activity_pos += 2;
-
             } else {
                 // A duplicated pointer will have the following two outer_fn arguments:
                 // (..., ptr, ptr, ...). We add the following llvm-ir to our __enzyme call:
@@ -161,7 +166,6 @@ fn match_args_from_caller_to_enzyme<'ll>(
                     args.push(next_outer_arg);
                     outer_pos += 1;
                 }
-
             }
         } else {
             // We do not differentiate with resprect to this argument.
@@ -172,7 +176,6 @@ fn match_args_from_caller_to_enzyme<'ll>(
     }
 }
 
-
 // On LLVM-IR, we can luckily declare __enzyme_ functions without specifying the input
 // arguments. We do however need to declare them with their correct return type.
 // We already figured the correct return type out in our frontend, when generating the outer_fn,
@@ -350,7 +353,14 @@ fn generate_enzyme_call<'ll>(
 
         let has_sret = has_sret(outer_fn);
         let outer_args: Vec<&llvm::Value> = get_params(outer_fn);
-        match_args_from_caller_to_enzyme(&cx, attrs.width, &mut args, &attrs.input_activity, &outer_args, has_sret);
+        match_args_from_caller_to_enzyme(
+            &cx,
+            attrs.width,
+            &mut args,
+            &attrs.input_activity,
+            &outer_args,
+            has_sret,
+        );
 
         let call = builder.call(enzyme_ty, ad_fn, &args, None);