4 ms·
But then SUB, DIV and RELU could be an HLOP as well, no?
by WithinReason 4y ago
But then SUB, DIV and RELU could be an HLOP as well, no?
- georgehotz 4y agoWe could have NEG instead of SUB, but with the constant folding it's a wash. DIV is already an HLOP with reciprocal (used to use POW, but that was slower. And what would you implement RELU in terms of?
- WithinReason 4y agomax(0,x)
- georgehotz 4y agoThat's a ReduceOp right now, more annoying to reason about than a UnaryOp. But in the limit, yea. Or add an elementwise BinaryOp for max. Submit a PR if you can improve something!