zkvmc-jit 0.0.1

Jit runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use std::ptr::NonNull;

use rustc_hash::FxHashMap;
use zkvmc_context::{Compiler, Context, JitFn};

/// A JIT cache that stores compiled functions.
pub struct JitCache<C> {
    inner: C,
    cache: FxHashMap<u32, JitFn>,
}

impl<C> JitCache<C> {
    pub fn new(inner: C) -> Self {
        Self {
            inner,
            cache: FxHashMap::default(),
        }
    }
}

impl<C> Compiler for JitCache<C>
where
    C: Compiler,
{
    fn compile(&mut self, ctx: NonNull<Context>) -> JitFn {
        *self
            .cache
            .entry(unsafe { ctx.as_ref().pc })
            .or_insert_with(|| self.inner.compile(ctx))
    }
}