Commit 54d89832 authored by Patrick McCormick's avatar Patrick McCormick
Browse files

New unified runtime abi for GPUs.

parent 2832d1ad

kitsune/runtimes/gpu/.gitignore

0 → 100644
+5 −0
Original line number Diff line number Diff line 
test

*.o

*.bc

*.ll

*.so

kitsune/runtimes/gpu/check-cuda.cc

0 → 100644
+6 −0
Original line number Diff line number Diff line 
#if __has_include("cuda.h")

#include"llvm-cuda.cc"

#else

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

#include"nocuda.cc"

#endif

kitsune/runtimes/gpu/check-hip.cc

0 → 100644
+6 −0
Original line number Diff line number Diff line 
#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

kitsune/runtimes/gpu/check-spirv.cc

0 → 100644
+6 −0
Original line number Diff line number Diff line 


#ifdef enable_spirv

#include"llvm-spirv.cc"

#else

#include"nospirv.cc"

#endif

kitsune/runtimes/gpu/gpu.cc

0 → 100644
+89 −0
Original line number Diff line number Diff line 
#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");

  }

}