fn get_generated_files()

in build2cmake/src/main.rs [366:394]

• 24 lines of code
• 3 McCabe index (conditional complexity)

fn get_generated_files(
    env: &Environment,
    build: &Build,
    target_dir: PathBuf,
    ops_id: Option<String>,
) -> Result<Vec<PathBuf>> {
    let mut all_set = FileSet::new();

    for backend in build.backends() {
        let set = match backend {
            Backend::Cuda | Backend::Rocm => {
                write_torch_ext_cuda(env, backend, build, target_dir.clone(), ops_id.clone())?
            }
            Backend::Metal => {
                write_torch_ext_metal(env, build, target_dir.clone(), ops_id.clone())?
            }
        };

        all_set.extend(set);
    }

    if build.general.universal {
        let set = write_torch_ext_universal(env, build, target_dir, ops_id)?;

        all_set.extend(set);
    }

    Ok(all_set.into_names())
}