[GENERAL] Various improvements:

* Sparse einsum in triton.ops.einsum
* Hacky support for fixed-tile-size atomic-add
* Various bugfixes in parser
This commit is contained in:
Philippe Tillet
2020-10-25 11:55:58 -07:00
parent 444907589d
commit 049ab989b5
16 changed files with 574 additions and 331 deletions

View File

@@ -523,7 +523,7 @@ void BinaryOp::RelationalOpTypeChecking() {
}
Convert();
}
type_ = ArithmType::New(T_INT);
type_ = ArithmType::New(T_BOOL);
Broadcast(this, lhs_, rhs_, type_);
}
@@ -538,7 +538,7 @@ void BinaryOp::EqualityOpTypeChecking() {
Error(this, "invalid operands to binary %s", tok_->str_.c_str());
Convert();
}
type_ = ArithmType::New(T_INT);
type_ = ArithmType::New(T_BOOL);
Broadcast(this, lhs_, rhs_, type_);
}
@@ -558,7 +558,7 @@ void BinaryOp::LogicalOpTypeChecking() {
::Type* rhsScalType = TryExtractScalarType(this, rhs_);
if (!lhsScalType->IsScalar() || !rhsScalType->IsScalar())
Error(this, "the operand should be arithmetic type or pointer");
type_ = ArithmType::New(T_INT);
type_ = ArithmType::New(T_BOOL);
Broadcast(this, lhs_, rhs_, type_);
}