cleanups

Author: ZuseZ4

compiler/rustc_builtin_macros/src/autodiff.rs

@@ -5,11 +5,11 @@
 //use crate::util::check_autodiff;
 
 use crate::errors;
-use rustc_ast::FnRetTy;
 use rustc_ast::expand::autodiff_attrs::{AutoDiffAttrs, DiffActivity, DiffMode};
 use rustc_ast::ptr::P;
 use rustc_ast::token::{Token, TokenKind};
 use rustc_ast::tokenstream::*;
+use rustc_ast::FnRetTy;
 use rustc_ast::{self as ast, FnHeader, FnSig, Generics, MetaItemKind, NestedMetaItem, StmtKind};
 use rustc_ast::{BindingAnnotation, ByRef};
 use rustc_ast::{Fn, ItemKind, PatKind, Stmt, TyKind, Unsafe};

compiler/rustc_codegen_llvm/src/back/write.rs

@@ -702,7 +702,6 @@ pub(crate) unsafe fn enzyme_ad(
     diag_handler: &DiagCtxt,
     item: AutoDiffItem,
 ) -> Result<(), FatalError> {
-    dbg!("cg_llvm enzyme_ad");
     let autodiff_mode = item.attrs.mode;
     let rust_name = item.source;
     let rust_name2 = &item.target;
@@ -743,6 +742,7 @@ pub(crate) unsafe fn enzyme_ad(
     };
     let src_num_args = llvm::LLVMCountParams(src_fnc);
     let target_num_args = llvm::LLVMCountParams(target_fnc);
+    // A really simple check
     assert!(src_num_args <= target_num_args);
 
     // create enzyme typetrees
@@ -820,7 +820,6 @@ pub(crate) unsafe fn differentiate(
     _typetrees: FxHashMap<String, DiffTypeTree>,
     _config: &ModuleConfig,
 ) -> Result<(), FatalError> {
-    dbg!("cg_llvm differentiate");
     dbg!(&diff_items);
 
     let llmod = module.module_llvm.llmod();

compiler/rustc_codegen_llvm/src/lib.rs

@@ -268,8 +268,6 @@ impl WriteBackendMethods for LlvmCodegenBackend {
         typetrees: FxHashMap<String, Self::TypeTree>,
         config: &ModuleConfig,
     ) -> Result<(), FatalError> {
-        dbg!("cg_llvm autodiff");
-        dbg!("Differentiating {} functions", diff_fncs.len());
         unsafe { back::write::differentiate(module, cgcx, diff_fncs, typetrees, config) }
     }
 

compiler/rustc_codegen_llvm/src/llvm/ffi.rs

@@ -936,7 +936,6 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
         _ => panic!("Invalid return activity"),
     };
 
-    //assert!(ret_activity == CDIFFE_TYPE::DFT_CONSTANT || ret_activity == CDIFFE_TYPE::DFT_OUT_DIFF);
     let input_activity: Vec<CDIFFE_TYPE> = input_activity.iter().map(|&x| cdiffe_from(x)).collect();
 
     dbg!(&fnc);
@@ -971,7 +970,7 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
         input_activity.len(), // constant arguments
         type_analysis,        // type analysis struct
         primary_ret as u8,
-        diff_ret as u8,                   //0
+        diff_ret as u8,                           //0
         CDerivativeMode::DEM_ReverseModeCombined, // return value, dret_used, top_level which was 1
         1,                                        // vector mode width
         1,                                        // free memory

compiler/rustc_codegen_ssa/src/back/write.rs

@@ -1545,7 +1545,6 @@ fn start_executing_work<B: ExtraBackendMethods>(
                 }
 
                 Message::AddAutoDiffItems(mut items) => {
-                    dbg!("AddAutoDiffItems");
                     autodiff_items.append(&mut items);
                 }
 
@@ -2019,7 +2018,6 @@ impl<B: ExtraBackendMethods> OngoingCodegen<B> {
     }
 
     pub fn submit_autodiff_items(&self, items: Vec<AutoDiffItem>) {
-        dbg!("submit_autodiff_items");
         drop(self.coordinator.sender.send(Box::new(Message::<B>::AddAutoDiffItems(items))));
     }
 

compiler/rustc_monomorphize/src/partitioning.rs

@@ -258,10 +258,6 @@ where
         //if visibility == Visibility::Hidden && can_be_internalized {
         let autodiff_active =
             characteristic_def_id.map(|x| cx.tcx.autodiff_attrs(x).is_active()).unwrap_or(false);
-        if autodiff_active {
-            dbg!("place_mono_items: autodiff_active");
-            dbg!(&mono_item);
-        }
 
         if !autodiff_active && visibility == Visibility::Hidden && can_be_internalized {
             internalization_candidates.insert(mono_item);
@@ -1188,7 +1184,6 @@ fn collect_and_partition_mono_items(
                     }
                     _ => None,
                 });
-            println!("source: {:?}", source);
 
             source.map(|inst| {
                 println!("source_id: {:?}", inst.def_id());