Added initial gpu runtime

#if __has_include("cuda.h")

#include"llvm-cuda.cc"

#else

#warning "Couldn't find cuda, not building cuda support"

#include"nocuda.cc"

#endif

#if __has_include("hip/hip_runtime.h")

#include"llvm-hip.cc"

#else

#warning "Couldn't find hip, not building hip support"

#include"nohip.cc"

#endif

#ifdef enable_spirv

#include"llvm-spirv.cc"

#else

#include"nospirv.cc"

#endif

#include"gpu.h"

#include"llvm-cuda.h"

#include"llvm-hip.h"

#include"llvm-spirv.h"

#include<llvm/IR/Module.h>

#include<llvm/IRReader/IRReader.h>

#include<llvm/Support/SourceMgr.h>

#include<error.h>

#include<stdbool.h>

void err(const char* msg){

  return error(1, 1, "%s", msg);

}

typedef enum {

  none,

  spirv,

  hip,

  cuda

} runtime;  

runtime globalRuntime = none;

void *gpuManagedMalloc(size_t n){

	switch(globalRuntime){

		case hip:

			return hipManagedMalloc(n);

		case cuda:

			return cudaManagedMalloc(n);

		default:

			err("no spirv managed malloc");

	}	

	return NULL;

}

void initRuntime(){

  if(globalRuntime != none) return;

  if(initCUDA()) {

		globalRuntime = cuda;

		return;

	}

  if(initHIP()){

		globalRuntime = hip; 

		return;

	}

  if(initSPIRV()){

		globalRuntime = spirv; 

		return;

	}

	err("No gpu runtimes found, needed OpenCL with SPIRV support, HIP, or CUDA\n");

}

void* launchBCKernel(const char* bc, void** args, size_t n){

  llvm::LLVMContext C; 

  llvm::SMDiagnostic SMD; 

  llvm::StringRef sr(bc); 

  llvm::MemoryBufferRef mbr(sr, "kernelModRef"); 

  std::unique_ptr<llvm::Module> mod =

      parseIR(mbr, SMD, C);

  return launchKernel(*mod, args, n); 

}

void* launchKernel(llvm::Module& bc, void** args, size_t n){

  switch(globalRuntime){

    case spirv: 

      return launchSPIRVKernel(bc, args, n);

    case hip:

      return launchHIPKernel(bc, args, n);

    case cuda:

      return launchCUDAKernel(bc, args, n);

    default:

      err("Can't get kernel without valid runtime");

  }

  return NULL; 

}

void waitKernel(void* wait){

  switch(globalRuntime){

    case spirv:

      return waitSPIRVKernel(wait);

    case hip:

      return waitHIPKernel(wait);

    case cuda:

      return waitCUDAKernel(wait);

    default:

      err("Can't wait kernel without valid runtime");

  }

}

#include<stdint.h>

#include<stddef.h>

#ifdef __cplusplus

#include<llvm/IR/Module.h>

extern "C" {

void* launchKernel(llvm::Module& bc, void** args, size_t n); 

#endif

void *gpuManagedMalloc(size_t n); 

void initRuntime(); 

void* launchBCKernel(const char* bc, void** args, size_t n); 

void waitKernel(void* wait); 

#ifdef __cplusplus

}

#endif