[general] removed useless files and includes
This commit is contained in:
@@ -1,11 +1,9 @@
|
||||
#ifndef _TRITON_CODEGEN_ANALYSIS_AXES_H_
|
||||
#define _TRITON_CODEGEN_ANALYSIS_AXES_H_
|
||||
|
||||
#include <map>
|
||||
#include <set>
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
#include "triton/tools/graph.h"
|
||||
#include <map>
|
||||
#include <vector>
|
||||
|
||||
namespace triton{
|
||||
|
||||
|
@@ -37,11 +37,7 @@ private:
|
||||
ir::value *reassociate_ptr(ir::getelementptr_inst* pz, ir::builder &builder, std::map<ir::value*, cst_info> &offsets);
|
||||
|
||||
public:
|
||||
reassociate(analysis::align* align);
|
||||
void run(ir::module& module);
|
||||
|
||||
private:
|
||||
analysis::align* align_;
|
||||
};
|
||||
|
||||
}
|
||||
|
@@ -5,9 +5,6 @@
|
||||
#include "triton/ir/basic_block.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/type.h"
|
||||
#include <iostream>
|
||||
#include <numeric>
|
||||
#include <algorithm>
|
||||
|
||||
namespace triton {
|
||||
namespace codegen{
|
||||
|
@@ -3,12 +3,6 @@
|
||||
#include "triton/codegen/analysis/layout.h"
|
||||
#include "triton/codegen/analysis/allocation.h"
|
||||
#include "triton/codegen/analysis/liveness.h"
|
||||
#include "triton/codegen/transform/cts.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
#include "triton/ir/type.h"
|
||||
#include "triton/ir/value.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/utils.h"
|
||||
|
||||
namespace triton{
|
||||
|
@@ -1,12 +1,7 @@
|
||||
#include "triton/codegen/analysis/axes.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/utils.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/type.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/context_impl.h"
|
||||
#include "triton/ir/constant.h"
|
||||
#include "triton/driver/device.h"
|
||||
|
||||
|
||||
|
||||
|
@@ -1,5 +1,4 @@
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
#include <numeric>
|
||||
#include "triton/codegen/analysis/axes.h"
|
||||
#include "triton/codegen/analysis/align.h"
|
||||
|
@@ -1,15 +1,8 @@
|
||||
#include <iostream>
|
||||
#include <climits>
|
||||
#include <unordered_set>
|
||||
#include "triton/codegen/instructions.h"
|
||||
#include "triton/codegen/analysis/liveness.h"
|
||||
#include "triton/codegen/analysis/layout.h"
|
||||
#include "triton/codegen/transform/cts.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/value.h"
|
||||
#include "triton/ir/utils.h"
|
||||
|
||||
namespace triton{
|
||||
|
@@ -1,15 +1,10 @@
|
||||
#include <iostream>
|
||||
#include <algorithm>
|
||||
#include <numeric>
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/utils.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/codegen/analysis/layout.h"
|
||||
#include "triton/codegen/transform/cts.h"
|
||||
#include "triton/codegen/analysis/align.h"
|
||||
#include "triton/codegen/transform/coalesce.h"
|
||||
#include "triton/codegen/analysis/align.h"
|
||||
#include "triton/codegen/analysis/layout.h"
|
||||
|
||||
namespace triton {
|
||||
namespace codegen{
|
||||
|
@@ -1,12 +1,9 @@
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
#include "triton/codegen/transform/cts.h"
|
||||
#include "triton/codegen/instructions.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
#include "triton/ir/instructions.h"
|
||||
#include "triton/ir/type.h"
|
||||
|
||||
namespace triton {
|
||||
namespace codegen{
|
||||
|
@@ -1,9 +1,8 @@
|
||||
#include "triton/codegen/transform/dce.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/utils.h"
|
||||
#include "triton/codegen/transform/dce.h"
|
||||
#include <iostream>
|
||||
|
||||
namespace triton {
|
||||
namespace codegen{
|
||||
|
@@ -1,13 +1,10 @@
|
||||
#include <vector>
|
||||
#include <set>
|
||||
#include <algorithm>
|
||||
|
||||
#include "triton/codegen/analysis/liveness.h"
|
||||
#include "triton/codegen/analysis/layout.h"
|
||||
#include "triton/codegen/analysis/allocation.h"
|
||||
#include "triton/codegen/instructions.h"
|
||||
#include "triton/codegen/transform/membar.h"
|
||||
#include "triton/codegen/transform/cts.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
|
@@ -2,7 +2,7 @@
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/codegen/transform/peephole.h"
|
||||
#include <iostream>
|
||||
|
||||
namespace triton {
|
||||
namespace codegen{
|
||||
namespace transform{
|
||||
|
@@ -1,7 +1,5 @@
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
#include "triton/codegen/transform/reassociate.h"
|
||||
#include "triton/codegen/analysis/align.h"
|
||||
#include "triton/ir/module.h"
|
||||
#include "triton/ir/function.h"
|
||||
#include "triton/ir/basic_block.h"
|
||||
@@ -12,27 +10,6 @@ namespace triton {
|
||||
namespace codegen{
|
||||
namespace transform{
|
||||
|
||||
//inline Constant *get_gep_cst_offset(GetElementPtrInst *gep){
|
||||
// std::vector<Value*> idx_vals;
|
||||
// std::transform(gep->idx_begin(), gep->idx_end(),
|
||||
// std::back_inserter(idx_vals),
|
||||
// [](Value* x){ return x;});
|
||||
// if(idx_vals.size() > 1)
|
||||
// return nullptr;
|
||||
// Value *idx = idx_vals[0];
|
||||
// if(isa<Constant>(idx))
|
||||
// return idx;
|
||||
// if(Instruction *BinOp = is_bin_add(idx)){
|
||||
// Value *LHS = BinOp->getOperand(0);
|
||||
// Value *RHS = BinOp->getOperand(1);
|
||||
// if(Constant* Res = dyn_cast<Constant>(LHS))
|
||||
// return Res;
|
||||
// if(Constant* Res = dyn_cast<Constant>(RHS))
|
||||
// return Res;
|
||||
// }
|
||||
// return nullptr;
|
||||
//}
|
||||
|
||||
|
||||
inline ir::instruction* reassociate::is_bin_add(ir::value *x) {
|
||||
ir::binary_operator *bin_op = dynamic_cast<ir::binary_operator*>(x);
|
||||
@@ -141,10 +118,6 @@ ir::value *reassociate::reassociate_idx(ir::value *old_value,
|
||||
return new_value;
|
||||
}
|
||||
|
||||
reassociate::reassociate(analysis::align *align): align_(align)
|
||||
{ }
|
||||
|
||||
|
||||
/* run */
|
||||
void reassociate::run(ir::module &mod) {
|
||||
ir::builder &builder = mod.get_builder();
|
||||
|
@@ -214,7 +214,7 @@ std::unique_ptr<driver::module> function::make_bin(ir::module &module, driver::c
|
||||
codegen::transform::membar barriers(&liveness, &layouts, &allocation);
|
||||
codegen::transform::dce dce;
|
||||
codegen::transform::peephole peephole;
|
||||
codegen::transform::reassociate reassociate(&align);
|
||||
codegen::transform::reassociate reassociate;
|
||||
codegen::transform::coalesce coalesce(&align, &layouts);
|
||||
codegen::transform::cts cts;
|
||||
codegen::generator isel(&axes, &layouts, &align, &allocation, target.get(), opt.num_warps);
|
||||
|
@@ -16,8 +16,8 @@ void reduce1d(TYPE * X __noalias __readonly __aligned(16),
|
||||
const char *reduce2d =
|
||||
R"(
|
||||
void reduce2d(TYPE * X __noalias __readonly __aligned(16),
|
||||
TYPE * Y __noalias __writeonly __aligned(16),
|
||||
int S0, int S1) {
|
||||
TYPE * Y __noalias __writeonly __aligned(16),
|
||||
int S0, int S1) {
|
||||
int pid0 = get_program_id(0);
|
||||
int pid1 = get_program_id(1);
|
||||
int rs0[TS0] = pid0 * TS0 + 0 ... TS0;
|
||||
|
Reference in New Issue
Block a user