gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

attempting vectorization

Browse Source
This commit is contained in:
Philippe Tillet
2019-02-10 18:29:25 -05:00
parent 4a0736ce20
commit 3d07e909c6
3 changed files with 69 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
include/codegen/selection.h
View File

@@ -25,6 +25,7 @@ class tune;
typedef std::vector<llvm::Value*> indices_t;
struct distributed_axis {
size_t contiguous;
std::vector<llvm::Value*> values;
};
@@ -33,7 +34,7 @@ protected:
typedef std::vector<unsigned> shapes_t;
public:
tile(llvm::Type *ty, const shapes_t &shapes): shapes_(shapes){ }
tile(llvm::Type *ty, const shapes_t &shapes): ty_(ty), shapes_(shapes){ }
virtual void set_value(indices_t idx, llvm::Value *v) = 0;
virtual llvm::Value* get_value(indices_t idx) = 0;
@@ -69,7 +70,9 @@ private:
void init_indices();
public:
distributed_tile(llvm::Type *ty, const shapes_t& shapes, const axes_t &axes);
distributed_tile(llvm::Type *ty, const shapes_t& shapes, const axes_t &axes, llvm::IRBuilder<> &builder);
void set_vectorized_iteration() { vectorized_ = true; }
void unset_vectorized_iteration() { vectorized_ = false; }
void set_value(indices_t idx, llvm::Value *v);
llvm::Value* get_value(indices_t idx);
void for_each(std::function<void(indices_t)> fn);
@@ -78,6 +81,9 @@ private:
axes_t axes_;
indices_map_t indices_;
values_t values_;
size_t vector_size_;
llvm::IRBuilder<> &builder_;
bool vectorized_;
};
@@ -86,6 +92,9 @@ class selection{
typedef std::map<ir::value *, tile *> tmap_t;
private:
// utils
llvm::Type *make_vector_ty(llvm::Type *ty, size_t vector_size);
// LLVM conversions
llvm::Type* llvm_type(ir::type *ty, llvm::LLVMContext &ctx);
llvm::Value* llvm_value(ir::value *v, llvm::IRBuilder<> &builder);