|
1 | | -use crate::partitioning::UsageMap; |
2 | | -use rustc_middle::mir::mono::MonoItem; |
3 | | -use rustc_middle::bug; |
4 | 1 | use rustc_ast::expand::autodiff_attrs::{AutoDiffItem, DiffActivity}; |
| 2 | +use rustc_hir::def_id::LOCAL_CRATE; |
| 3 | +use rustc_middle::bug; |
| 4 | +use rustc_middle::mir::mono::MonoItem; |
| 5 | +use rustc_middle::ty::{self, Instance, Ty, TyCtxt}; |
5 | 6 | use rustc_symbol_mangling::symbol_name_for_instance_in_crate; |
6 | 7 | use tracing::{debug, trace}; |
7 | | -use rustc_middle::ty::{self, Instance, Ty, TyCtxt}; |
8 | | -use rustc_hir::def_id::LOCAL_CRATE; |
9 | 8 |
|
10 | | -fn adjust_activity_to_abi<'tcx>( |
11 | | - tcx: TyCtxt<'tcx>, |
12 | | - fn_ty: Ty<'tcx>, |
13 | | - da: &mut Vec<DiffActivity>, |
14 | | -) { |
| 9 | +use crate::partitioning::UsageMap; |
| 10 | + |
| 11 | +fn adjust_activity_to_abi<'tcx>(tcx: TyCtxt<'tcx>, fn_ty: Ty<'tcx>, da: &mut Vec<DiffActivity>) { |
15 | 12 | if !matches!(fn_ty.kind(), ty::FnDef(..)) { |
16 | 13 | bug!("expected fn def for autodiff, got {:?}", fn_ty); |
17 | 14 | } |
@@ -61,7 +58,11 @@ fn adjust_activity_to_abi<'tcx>( |
61 | 58 | } |
62 | 59 | } |
63 | 60 |
|
64 | | -pub(crate) fn find_autodiff_source_functions<'tcx>(tcx: TyCtxt<'tcx>, usage_map: &UsageMap<'tcx>, autodiff_mono_items: Vec<(&MonoItem<'tcx>,&Instance<'tcx>)>) -> Vec<AutoDiffItem> { |
| 61 | +pub(crate) fn find_autodiff_source_functions<'tcx>( |
| 62 | + tcx: TyCtxt<'tcx>, |
| 63 | + usage_map: &UsageMap<'tcx>, |
| 64 | + autodiff_mono_items: Vec<(&MonoItem<'tcx>, &Instance<'tcx>)>, |
| 65 | +) -> Vec<AutoDiffItem> { |
65 | 66 | let mut autodiff_items: Vec<AutoDiffItem> = vec![]; |
66 | 67 | for (item, instance) in autodiff_mono_items { |
67 | 68 | let target_id = instance.def_id(); |
|
0 commit comments