browsermt/bergamot-translator
1
1
Fork 0
mirror of https://github.com/browsermt/bergamot-translator.git synced 2024-09-11 05:35:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

JS/WASM: Re-enable importing optimized gemm module for (#336)

Browse Source 
- Re-enabled the code that imports optimized gemm module
   for wasm when available
This commit is contained in:
Abhishek Aggarwal 2022-02-07 16:55:31 +01:00 committed by GitHub
parent f6d9233dc4
commit 6b2a855234
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
wasm/import-gemm-module.js
View File

@ -14,28 +14,22 @@ function createWasmGemm() {
"int8_select_columns_of_b": "int8SelectColumnsOfBFallback"
};
// ToDo: Activate the if code and remove else code once optimized gemm can work without shared array buffer.
if (0) {
// Name of the optimized gemm implementation.
const OPTIMIZED_GEMM = "mozIntGemm";
// Name of the optimized gemm implementation.
const OPTIMIZED_GEMM = "mozIntGemm";
const optimizedGemmModule = WebAssembly[OPTIMIZED_GEMM];
if (!optimizedGemmModule) {
return fallbackGemm(GEMM_TO_FALLBACK_FUNCTIONS_MAP);
}
const optimizedGemmModuleExports = new WebAssembly.Instance(optimizedGemmModule(), {"": {memory: wasmMemory}}).exports;
for (let key in GEMM_TO_FALLBACK_FUNCTIONS_MAP) {
if (!optimizedGemmModuleExports[key]) {
return fallbackGemm(GEMM_TO_FALLBACK_FUNCTIONS_MAP);
}
}
console.log(`Using optimized gemm (${OPTIMIZED_GEMM}) implementation`);
return optimizedGemmModuleExports;
}
else {
const optimizedGemmModule = WebAssembly[OPTIMIZED_GEMM];
if (!optimizedGemmModule) {
return fallbackGemm(GEMM_TO_FALLBACK_FUNCTIONS_MAP);
}
const optimizedGemmModuleExports = new WebAssembly.Instance(optimizedGemmModule(), {"": {memory: wasmMemory}}).exports;
for (let key in GEMM_TO_FALLBACK_FUNCTIONS_MAP) {
if (!optimizedGemmModuleExports[key]) {
return fallbackGemm(GEMM_TO_FALLBACK_FUNCTIONS_MAP);
}
}
console.log(`Using optimized gemm (${OPTIMIZED_GEMM}) implementation`);
return optimizedGemmModuleExports;
}
// Return the fallback gemm implementation.