[jit] Cleanup the usage of DISABLE_JIT a bit so it encompasses the whole mini-codegen...