d8d6b715c8
Improved codegen for the Ampere GPUs.
* Make the layout pass recognize the multistage pipelined pattern.
* Now the pipeline pass can automate the multistage pipelining transformation.
* Remove extra barriers (from the prefetch pass & WAR) on Ampere.
* Update the code generator (generator.cc) to make Triton generate n-buffered shared memory loads/stores.
27 lines
556 B
C++
27 lines
556 B
C++
#ifndef TRITON_INCLUDE_TRITON_CODEGEN_TRANSFORM_PREFETCH_H
|
|
#define TRITON_INCLUDE_TRITON_CODEGEN_TRANSFORM_PREFETCH_H
|
|
|
|
#include <set>
|
|
|
|
// forward dclaration
|
|
namespace triton::ir{
|
|
class module;
|
|
class value;
|
|
}
|
|
|
|
namespace triton::codegen {
|
|
class target;
|
|
}
|
|
|
|
namespace triton::codegen::transform {
|
|
class prefetch {
|
|
target* tgt_;
|
|
std::set<ir::value*> prefetched_vals_;
|
|
public:
|
|
prefetch(target *tgt) : tgt_(tgt) {}
|
|
void run(ir::module &module);
|
|
bool is_prefetched(ir::value* v) { return prefetched_vals_.find(v) != prefetched_vals_.end(); }
|
|
};
|
|
}
|
|
|
|
#endif |