update autodiff flags
This commit is contained in:
parent
161a4bf6ff
commit
e2d250c3f6
11 changed files with 204 additions and 76 deletions
|
@ -405,7 +405,8 @@ fn generate_lto_work<B: ExtraBackendMethods>(
|
|||
B::run_fat_lto(cgcx, needs_fat_lto, import_only_modules).unwrap_or_else(|e| e.raise());
|
||||
if cgcx.lto == Lto::Fat && !autodiff.is_empty() {
|
||||
let config = cgcx.config(ModuleKind::Regular);
|
||||
module = unsafe { module.autodiff(cgcx, autodiff, config).unwrap() };
|
||||
module =
|
||||
unsafe { module.autodiff(cgcx, autodiff, config).unwrap_or_else(|e| e.raise()) };
|
||||
}
|
||||
// We are adding a single work item, so the cost doesn't matter.
|
||||
vec![(WorkItem::LTO(module), 0)]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue