1 | #include <string> |
---|---|
2 | #include "triton/driver/dispatch.h" |
3 | |
4 | namespace llvm{ |
5 | class Module; |
6 | } |
7 | |
8 | namespace triton{ |
9 | namespace driver{ |
10 | |
11 | void init_llvm(); |
12 | std::string path_to_ptxas(int& version); |
13 | std::string llir_to_ptx(llvm::Module* module, int cc, int version); |
14 | std::string ptx_to_cubin(const std::string& ptx, const std::string& ptxas_path, int cc); |
15 | CUmodule ptx_to_cumodule(const std::string& ptx, int cc); |
16 | std::string llir_to_amdgpu(llvm::Module* module, const std::string& proc); |
17 | hipModule_t amdgpu_to_hipmodule(const std::string& path); |
18 | |
19 | } |
20 | } |
21 |