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

[LANG] Now requiring tiles have power of 2 number of elements

Browse Source
This commit is contained in:
Philippe Tillet
2021-02-04 16:13:10 -05:00
parent cc84a476a3
commit b8a52c70c9
4 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/lang/ast.cc
View File

@@ -448,7 +448,10 @@ void BinaryOp::RangeOpTypeChecking() {
int len = static_cast<int>(end - begin);
if(len < 0)
Error(this, "range cannot be negative");
type_ = TileType::New(TileType::ShapeInt{len}, lhs_->Type());
TileType* ret = TileType::New(TileType::ShapeInt{len}, lhs_->Type());
if(!ret->CheckPow2NumEl())
Error(this, "range must have power of 2 number of elements");
type_ = ret;
}
void BinaryOp::MaskedDerefOpTypeChecking() {
@@ -751,6 +754,8 @@ void UnaryOp::CastOpTypeChecking() {
if(type_->IsScalar() && operandType->ToTile()->NumEle() != 1)
Error(this, "tile with more than one element cannot be casted to scalar");
if(type_->IsTile() && operandType->IsTile()){
if(!type_->ToTile()->CheckPow2NumEl())
Error(this, "tile must have power of 2 number of elements");
auto operandShape = operandType->ToTile()->Shape();
auto shape = type_->ToTile()->Shape();
// this is a shape downcast