portal/waffle
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Parallelize with rayon

Browse source
This commit is contained in:
Chris Fallin 2022-12-01 10:02:42 -08:00
parent ae091d5319
commit f83420bbf7
No known key found for this signature in database
GPG key ID: 31649E4FE65EB465
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/backend/mod.rs
View file

@ -6,6 +6,7 @@ use crate::ir::{ExportKind, FuncDecl, FunctionBody, ImportKind, Module, Type, Va
use crate::passes::rpo::RPO; use crate::passes::rpo::RPO;
use crate::Operator; use crate::Operator;
use anyhow::Result; use anyhow::Result;
use rayon::prelude::*;
use std::borrow::Cow; use std::borrow::Cow;
pub mod stackify; pub mod stackify;
@ -689,10 +690,19 @@ pub fn compile(module: &Module<'_>) -> anyhow::Result<Vec<u8>> {
into_mod.section(&elem); into_mod.section(&elem);
let mut code = wasm_encoder::CodeSection::new(); let mut code = wasm_encoder::CodeSection::new();
for (func, func_decl) in module.funcs().skip(num_func_imports) { let bodies = module
let body = func_decl.body().unwrap(); .funcs()
log::debug!("Compiling {}", func); .skip(num_func_imports)
let body = WasmFuncBackend::new(body)?.compile()?; .collect::<Vec<_>>()
.par_iter()
.map(|(func, func_decl)| -> Result<wasm_encoder::Function> {
let body = func_decl.body().unwrap();
log::debug!("Compiling {}", func);
WasmFuncBackend::new(body)?.compile()
})
.collect::<Result<Vec<_>>>()?;
for body in bodies {
code.function(&body); code.function(&body);
} }
into_mod.section(&code); into_mod.section(&code);