mirror of
https://github.com/mii443/wasmer.git
synced 2025-08-28 11:19:25 +00:00
79 lines
2.6 KiB
Rust
79 lines
2.6 KiB
Rust
use std::sync::Arc;
|
|
use wasmer::{FunctionMiddlewareGenerator, Store};
|
|
use wasmer_compiler::CompilerConfig;
|
|
use wasmer_engine::Engine;
|
|
#[cfg(feature = "test-jit")]
|
|
use wasmer_engine_jit::JIT;
|
|
#[cfg(feature = "test-native")]
|
|
use wasmer_engine_native::Native;
|
|
|
|
pub fn get_compiler(canonicalize_nans: bool) -> impl CompilerConfig {
|
|
cfg_if::cfg_if! {
|
|
if #[cfg(any(
|
|
all(feature = "test-llvm", any(feature = "test-cranelift", feature = "test-singlepass")),
|
|
all(feature = "test-cranelift", feature = "test-singlepass")
|
|
))] {
|
|
compile_error!("Only one compiler can be selected")
|
|
} else if #[cfg(feature = "test-cranelift")] {
|
|
let mut compiler = wasmer_compiler_cranelift::Cranelift::new();
|
|
compiler.canonicalize_nans(canonicalize_nans);
|
|
compiler.enable_verifier();
|
|
compiler
|
|
} else if #[cfg(feature = "test-llvm")] {
|
|
let mut compiler = wasmer_compiler_llvm::LLVM::new();
|
|
compiler.canonicalize_nans(canonicalize_nans);
|
|
compiler.enable_verifier();
|
|
compiler
|
|
} else if #[cfg(feature = "test-singlepass")] {
|
|
let mut compiler = wasmer_compiler_singlepass::Singlepass::new();
|
|
compiler.canonicalize_nans(canonicalize_nans);
|
|
compiler.enable_verifier();
|
|
compiler
|
|
} else {
|
|
compile_error!("No compiler chosen for the tests")
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "test-jit")]
|
|
pub fn get_engine() -> impl Engine {
|
|
let compiler_config = get_compiler(false);
|
|
JIT::new(&compiler_config).engine()
|
|
// let mut compiler_config = get_compiler(false);
|
|
// Native::new(&mut compiler_config).engine()
|
|
}
|
|
#[cfg(feature = "test-native")]
|
|
pub fn get_engine() -> impl Engine {
|
|
let mut compiler_config = get_compiler(false);
|
|
Native::new(&mut compiler_config).engine()
|
|
}
|
|
|
|
pub fn get_store() -> Store {
|
|
Store::new(&get_engine())
|
|
}
|
|
|
|
pub fn get_store_with_middlewares<I: Iterator<Item = Arc<dyn FunctionMiddlewareGenerator>>>(
|
|
middlewares: I,
|
|
) -> Store {
|
|
let mut compiler_config = get_compiler(false);
|
|
for x in middlewares {
|
|
compiler_config.push_middleware(x);
|
|
}
|
|
#[cfg(feature = "test-jit")]
|
|
let engine = JIT::new(&compiler_config).engine();
|
|
#[cfg(feature = "test-native")]
|
|
let engine = Native::new(&mut compiler_config).engine();
|
|
Store::new(&engine)
|
|
}
|
|
|
|
#[cfg(feature = "test-jit")]
|
|
pub fn get_headless_store() -> Store {
|
|
Store::new(&JIT::headless().engine())
|
|
// Store::new(&Native::headless().engine())
|
|
}
|
|
|
|
#[cfg(feature = "test-native")]
|
|
pub fn get_headless_store() -> Store {
|
|
Store::new(&Native::headless().engine())
|
|
}
|