858dec8372
* Add cache modifier to tl.load * Add comment to cache_modifier * Remove force_nc_cache * Update test
43 lines
748 B
C++
43 lines
748 B
C++
#ifndef _TRITON_CODEGEN_PASS_H_
|
|
#define _TRITON_CODEGEN_PASS_H_
|
|
|
|
|
|
#include <memory>
|
|
|
|
namespace llvm{
|
|
class Module;
|
|
class LLVMContext;
|
|
}
|
|
|
|
namespace triton{
|
|
|
|
namespace codegen {
|
|
class target;
|
|
}
|
|
|
|
namespace ir{
|
|
class module;
|
|
}
|
|
namespace driver{
|
|
class device;
|
|
class module;
|
|
class kernel;
|
|
}
|
|
}
|
|
|
|
namespace triton{
|
|
namespace codegen{
|
|
|
|
// TODO:
|
|
// There should be a proper pass manager there!
|
|
std::unique_ptr<llvm::Module> add_passes_to_emit_bin(ir::module &ir, llvm::LLVMContext& ctx,
|
|
codegen::target* target,
|
|
int sm, int num_warps,
|
|
int num_stages, int &shared_static);
|
|
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|