jit.c File Reference

#include "serene/jit/jit.h"
#include <llvm-c/Core.h>
#include <llvm-c/Error.h>
#include <llvm-c/LLJIT.h>
#include <llvm-c/Support.h>
#include <llvm-c/Target.h>
#include <stdio.h>
#include "serene/rt/context.h"
#include "serene/rt/engine.h"
#include "serene/rt/mm/interface.h"
#include "serene/utils.h"
#include <string.h>

Include dependency graph for jit.c:

Go to the source code of this file.

Functions
static int	handle_error (LLVMErrorRef err)
srn_jit_t *	srn_jit_make (srn_mm_t *mm)
int	srn_jit_make_and_initialize (srn_jit_t *result)
int	srn_jit_shutdown (srn_jit_t *jit)
srn_error_t *	srn_jit_add_module (srn_context_t *ctx, srn_jit_module_t *module)
	Add a module to the jit compiler.

Function Documentation

handle_error()

static int handle_error ( LLVMErrorRef err )

inlinestatic

Definition at line 36 of file jit.c.

                                                 {
  // TODO(lxsameer): Use the runtime diagnostic system here.
  char *err_msg = LLVMGetErrorMessage(err);
  UNUSED(fprintf(stderr, "Error: %s\n", err_msg));
  LLVMDisposeErrorMessage(err_msg);
  return 1;
}

Here is the caller graph for this function:

srn_jit_add_module()

srn_error_t * srn_jit_add_module	(	srn_context_t *	ctx,
		srn_jit_module_t *	module )

Add a module to the jit compiler.

Definition at line 83 of file jit.c.

                                                                              {
  PANIC_IF_NULL(ctx);
  PANIC_IF_NULL(module);
 
  LLVMOrcLLJITRef j = ctx->engine->jit->llvm_jit;
 
  LLVMOrcJITDylibRef jd = LLVMOrcLLJITGetMainJITDylib(j);
  LLVMErrorRef err = LLVMOrcLLJITAddLLVMIRModule(j, jd, module->tsm);
  srn_error_t *srn_err = nullptr;
 
  if (err != nullptr) {
    // If adding the ThreadSafeModule fails then we need to clean it up
    // ourselves. If adding it succeeds the JIT will manage the memory.
    // LLVMOrcDisposeThreadSafeModule(module);
 
    // LLVM owns the string message, but we want to own it
    char *jit_err_msg = LLVMGetErrorMessage(err);
    char *msg = srn_copy_bytes(ctx, jit_err_msg, strlen(jit_err_msg), alignof(char));
    srn_err = srn_errors_make(ctx, FAILED_TO_ADD_MODULE, msg);
    LLVMDisposeErrorMessage(jit_err_msg);
  }
  return srn_err;
}

Here is the call graph for this function:

srn_jit_make()

srn_jit_t * srn_jit_make

(

srn_mm_t *

mm

)

Definition at line 44 of file jit.c.

                                      {
  PANIC_IF_NULL(mm);
  srn_jit_t *jit = srn_mm_immortal_allocate(mm, srn_jit_t);
  PANIC_IF_NULL(jit);
  jit->builder = nullptr;
  jit->llvm_jit = nullptr;
  return jit;
}

Here is the caller graph for this function:

srn_jit_make_and_initialize()

int srn_jit_make_and_initialize

(

srn_jit_t *

result

)

Definition at line 53 of file jit.c.

                                                   {
  PANIC_IF_NULL(result);
 
  int error = 0;
  // TODO(lxsameer): Check for the target platform and setup the codegen
  // and printer accordingly. Only applies to when we want to
  // link the jit library into the target binary.
 
  // Initialize native target codegen and asm printer.
  LLVMInitializeNativeTarget();
  LLVMInitializeNativeAsmPrinter();
 
  LLVMErrorRef err = nullptr;
  err = LLVMOrcCreateLLJIT(&result->llvm_jit, result->builder);
  if (err) {
    error = handle_error(err);
    LLVMShutdown();
    return error;
  }
  return 0;
}

Here is the call graph for this function:

srn_jit_shutdown()

int srn_jit_shutdown

(

srn_jit_t *

jit

)

Definition at line 75 of file jit.c.

                                     {
  LLVMErrorRef err = LLVMOrcDisposeLLJIT(jit->llvm_jit);
  if (err) {
    return handle_error(err);
  }
  return 0;
}

Here is the call graph for this function:

Here is the caller graph for this function: