Binary Shift operators - Githubissues

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

Apache License 2.0

475 stars 75 forks source link

Binary Shift operators #10034

Open VirdhatchaniKN opened 4 months ago

VirdhatchaniKN commented 4 months ago

Master issue : #9702

Op Requirement :

ElemwiseBinary/ELEMWISE_BINARY_LEFT_SHIFT
ElemwiseBinary/ELEMWISE_BINARY_RIGHT_SHIFT

Issues :

Initial Implementation done in #8636 This implementation requires the involvement of a lot of ops, which may lead to performance issues.

KalaivaniMCW commented 1 month ago

Hi @rtawfik01 @rdjogoTT

Is it possible to support shift operation for two tensors (like it is available in bitwise operation)? At present, the second operand for shift operation takes only a single integer parameter

Bitwise ops: