[FRONTEND] Added on-disk cache for compiled kernels (#287)

2021-09-18 22:48:26 -07:00
parent bd855ac13d
commit 6e5b0b4301
5 changed files with 235 additions and 81 deletions
--- a/include/triton/driver/llvm.h
+++ b/include/triton/driver/llvm.h
@@ -10,6 +10,7 @@ namespace driver{

 void init_llvm();
 std::string llir_to_ptx(llvm::Module* module, int cc, int version);
+std::string ptx_to_cubin(const std::string& ptx, int cc);
 CUmodule ptx_to_cumodule(const std::string& ptx, int cc);
 std::string llir_to_amdgpu(llvm::Module* module, const std::string& proc);
 hipModule_t amdgpu_to_hipmodule(const std::string& path);