-
Notifications
You must be signed in to change notification settings - Fork 46
Open
Labels
good first issueGood for newcomersGood for newcomers
Description
The following elementwise binary operations need an implementation in PTensorToLinalg(.cpp) (see array-API spec for the expected behavior of the operations):
- ptensor::ATAN2
- ptensor::LOGADDEXP
- ptensor::LSHIFT
- ptensor::MATMUL
- ptensor::TRUE_DIVIDE
- ptensor::BITWISE_AND
- ptensor::BITWISE_LEFT_SHIFT
- ptensor::BITWISE_OR
- ptensor::BITWISE_RIGHT_SHIFT
- ptensor::BITWISE_XOR
- ptensor::EQUAL
- ptensor::GREATER
- ptensor::GREATER_EQUAL
- ptensor::LESS
- ptensor::LESS_EQUAL
- ptensor::LOGICAL_AND
- ptensor::LOGICAL_OR
- ptensor::LOGICAL_XOR
- ptensor::NOT_EQUAL
See also Operation Details->Elementwise Operations in the RFC.
Reference implementations can be found in the TOSA dialect and/or on main (numba_dpcomp/numba_dpcomp/mlir/numpy/funcs.py [Python]).
It is ok to let initial implementations operate on default PTensorTypes only, e.g. ignore device and distribution attributes of input tensors.
Metadata
Metadata
Assignees
Labels
good first issueGood for newcomersGood for newcomers