[jit] Add some micro-optimizations to speed up the AOT compiler.