WasmExecutor.h

Go to the documentation of this file.

#ifndef HALIDE_WASM_EXECUTOR_H
#define HALIDE_WASM_EXECUTOR_H
 
/** \file
 *
 * Support for running Halide-compiled Wasm code in-process.
 * Bindings for parameters, extern calls, etc. are established and the
 * Wasm code is executed. Allows calls to realize to work
 * exactly as if native code had been run, but via a JavaScript/Wasm VM.
 * Currently, only the WABT interpreter is supported.
 */
 
#include "Argument.h"
#include "Parameter.h"
#include "Type.h"
 
#include <map>
#include <string>
#include <vector>
 
namespace Halide {
 
struct JITExtern;
struct Target;
 
namespace Internal {
 
struct JITModule;
struct WasmModuleContents;
 
/** Handle to compiled wasm code which can be called later. */
struct WasmModule {
    Internal::IntrusivePtr<WasmModuleContents> contents;
 
    /** If the given target can be executed via the wasm executor, return true. */
    static bool can_jit_target(const Target &target);
 
    /** Compile generated wasm code with a set of externs. */
    static WasmModule compile(
        const Module &module,
        const std::vector<Argument> &arguments,
        const std::string &fn_name,
        const std::map<std::string, JITExtern> &externs,
        const std::vector<JITModule> &extern_deps);
 
    /** Run generated previously compiled wasm code with a set of arguments. */
    int run(const void *const *args);
};
 
}  // namespace Internal
}  // namespace Halide
 
#endif  // HALIDE_WASM_EXECUTOR_H