[CORE] Fixed several issues that arose in the development of the

torch-blocksparse package:

* Now using warp shuffle in reductions when possible
* Various bugfixes in layout inference
* Added INFINITY, exponential and select
* Better error messages for unimplemented constructs
This commit is contained in:
Philippe Tillet
2020-03-31 18:55:31 -04:00
committed by Philippe Tillet
parent ac26fbdc1f
commit 3304629de9
33 changed files with 374 additions and 135 deletions

View File

@@ -655,6 +655,9 @@ void UnaryOp::TypeChecking() {
case Token::REDUCE:
return ReduceOpTypeChecking();
case Token::EXP:
return IntrinsicOpTypeChecking();
default:
assert(false);
}
@@ -769,6 +772,10 @@ void UnaryOp::CastOpTypeChecking() {
}
}
void UnaryOp::IntrinsicOpTypeChecking() {
type_ = ScalarOrLikeTile(operand_, ArithmType::New(T_FLOAT));
}
/*
* Transposition Operator
*/