gpu host code generation

This commit is contained in:
Manuel Drehwald
2025-07-02 16:36:30 -07:00
parent 5958ebe829
commit 4a1a5a4295
4 changed files with 464 additions and 7 deletions

View File

@@ -654,6 +654,7 @@ pub(crate) fn run_pass_manager(
// We then run the llvm_optimize function a second time, to optimize the code which we generated
// in the enzyme differentiation pass.
let enable_ad = config.autodiff.contains(&config::AutoDiff::Enable);
let enable_gpu = config.offload.contains(&config::Offload::Enable);
let stage = if thin {
write::AutodiffStage::PreAD
} else {