Skip to content

Commit

Permalink
Update JAX LAX compatibility table. (#36)
Browse files Browse the repository at this point in the history
  • Loading branch information
@balancap
balancap authored Sep 27, 2023
1 parent 51e90a0 commit e418dac
Showing 1 changed file with 48 additions and 48 deletions.
96 changes: 48 additions & 48 deletions docs/operations.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,17 +9,17 @@
| `abs` | :white_check_mark: | :white_check_mark: | |
| `add` | :white_check_mark: | :white_check_mark: | |
| `acos` | :x: | :x: | Not directly available in Graphcore Poplibs |
| `approx_max_k` | :x: | :x: | |
| `approx_min_k` | :x: | :x: | |
| `argmax` | :x: | :x: | |
| `argmin` | :x: | :x: | |
| `approx_max_k` | :x: | n/a | |
| `approx_min_k` | :x: | n/a | |
| `argmax` | :x: | n/a | |
| `argmin` | :x: | n/a | |
| `asin` | :white_check_mark: | :white_check_mark: | |
| `atan` | :white_check_mark: | :white_check_mark: | |
| `atan2` | :white_check_mark: | :white_check_mark: | |
| `batch_matmul` | :x: | :x: | See `dot_general` |
| `bessel_i0e` | :x: | :x: | |
| `bessel_i1e` | :x: | :x: | |
| `betainc` | :x: | :x: | |
| `batch_matmul` | :x: | n/a | See `dot_general` |
| `bessel_i0e` | :x: | n/a | |
| `bessel_i1e` | :x: | n/a | |
| `betainc` | :x: | n/a | |
| `bitcast_convert_type` | :white_check_mark: | :white_check_mark: | Only same size dtype supported. |
| `bitwise_not` | :white_check_mark: | :x: | |
| `bitwise_and` | :white_check_mark: | :x: | |
Expand All @@ -31,14 +31,14 @@
| `cbrt` | :white_check_mark: | :white_check_mark: | |
| `ceil` | :white_check_mark: | :white_check_mark: | |
| `clamp` | :white_check_mark: | :x: | |
| `collapse` | :x: | :x: | |
| `complex` | :x: | :x: | |
| `concatenate` | :white_check_mark: | :x: | |
| `conj` | :x: | :x: | |
| `conv` | :x: | :x: | |
| `collapse` | :white_check_mark: | :white_check_mark: | |
| `complex` | :x: | n/a | Complex not supported in IPU XLA backend |
| `concatenate` | :white_check_mark: | n/a | |
| `conj` | :x: | n/a | Complex not supported in IPU XLA backend |
| `conv` | :x: | n/a | |
| `convert_element_type` | :white_check_mark: | :x: | |
| `conv_general_dilated` | :x: | :x: | |
| `conv_transpose` | :x: | :x: | |
| `conv_general_dilated` | :x: | n/a | |
| `conv_transpose` | :x: | n/a | |
| `cos` | :white_check_mark: | :white_check_mark: | |
| `cosh` | :white_check_mark: | :white_check_mark: | |
| `cummax` | :x: | :x: | |
Expand All @@ -47,66 +47,66 @@
| `cumsum` | :x: | :x: | |
| `digamma` | :x: | :x: | |
| `div` | :white_check_mark: | :white_check_mark: | |
| `dot` | :white_check_mark: | :x: | |
| `dot_general` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `dynamic_slice` | :x: | :x: | |
| `dynamic_update_slice` | :x: | :x: | |
| `eq` | :white_check_mark: | :x: | |
| `dot` | :white_check_mark: | n/a | |
| `dot_general` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `dynamic_slice` | :x: | n/a | |
| `dynamic_update_slice` | :x: | n/a | |
| `eq` | :white_check_mark: | n/a | |
| `erf` | :white_check_mark: | :white_check_mark: | |
| `erfc` | :x: | :x: | Not directly available in Graphcore Poplibs |
| `erf_inv` | :x: | :x: | Not directly available in Graphcore Poplibs |
| `exp` | :white_check_mark: | :x: | |
| `expand_dims` | :white_check_mark: | :x: | |
| `expm1` | :white_check_mark: | :x: | |
| `fft` | :x: | :x: | |
| `floor` | :white_check_mark: | :x: | |
| `full` | :question: | :x: | |
| `full_like` | :question: | :x: | |
| `gather` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `ge` | :white_check_mark: | :x: | |
| `gt` | :white_check_mark: | :x: | |
| `exp` | :white_check_mark: | :white_check_mark: | |
| `expand_dims` | :white_check_mark: | :white_check_mark: | |
| `expm1` | :white_check_mark: | :white_check_mark: | |
| `fft` | :x: | n/a | |
| `floor` | :white_check_mark: | :white_check_mark: | |
| `full` | :question: | n/a | |
| `full_like` | :question: | n/a | |
| `gather` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `ge` | :white_check_mark: | n/a | |
| `gt` | :white_check_mark: | n/a | |
| `igamma` | :x: | :x: | |
| `igammac` | :x: | :x: | |
| `imag` | :x: | :x: | |
| `index_in_dim` | :x: | :x: | |
| `index_take` | :x: | :x: | |
| `index_in_dim` | :x: | n/a | |
| `index_take` | :x: | n/a | |
| `iota` | :white_check_mark: | n/a | |
| `is_finite` | :white_check_mark: | :x: | |
| `le` | :white_check_mark: | :x: | |
| `lt` | :white_check_mark: | :x: | |
| `is_finite` | :white_check_mark: | n/a | |
| `le` | :white_check_mark: | n/a | |
| `lt` | :white_check_mark: | n/a | |
| `lgamma` | :x: | :x: | |
| `log` | :white_check_mark: | :white_check_mark: | |
| `log1p` | :white_check_mark: | :white_check_mark: | |
| `logistic` | :x: | :x: | |
| `max` | :white_check_mark: | :white_check_mark: | |
| `min` | :white_check_mark: | :white_check_mark: | |
| `mul` | :white_check_mark: | :white_check_mark: | |
| `ne` | :white_check_mark: | :x: | |
| `ne` | :white_check_mark: | n/a | |
| `neg` | :white_check_mark: | :white_check_mark: | |
| `nextafter` | :x: | :x: | |
| `pad` | :x: | :x: | |
| `polygamma` | :x: | :x: | |
| `pow` | :white_check_mark: | :white_check_mark: | |
| `real` | :x: | :x: | |
| `real` | :x: | n/a | Complex not supported in IPU XLA backend |
| `reciprocal` | :white_check_mark: | :x: | |
| `reduce` | :white_check_mark: | :x: | |
| `reduce` | :white_check_mark: | n/a | |
| `reshape` | :white_check_mark: | :white_check_mark: | `dimensions` argument not supported. |
| `rem` | :white_check_mark: | :white_check_mark: | |
| `rev` | :x: | :x: | |
| `round` | :white_check_mark: | :white_check_mark: | |
| `rsqrt` | :white_check_mark: | :white_check_mark: | |
| `scatter` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `scatter_add` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `scatter_max` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `scatter_min` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `scatter_mul` | :white_check_mark: | :x: | Limited set of configurations. See below. |
| `scatter` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `scatter_add` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `scatter_max` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `scatter_min` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `scatter_mul` | :white_check_mark: | n/a | Limited set of configurations. See below. |
| `select` | :white_check_mark: | :x: | |
| `shift_left` | :white_check_mark: | :x: | |
| `shift_right_arithmetic`| :white_check_mark: | :x: | |
| `shift_right_logical` | :white_check_mark: | :x: | |
| `slice` | :white_check_mark: | :x: | |
| `slice_in_dim` | :white_check_mark: | :x: | |
| `sign` | :white_check_mark: | :x: | |
| `slice` | :white_check_mark: | n/a | |
| `slice_in_dim` | :white_check_mark: | n/a | |
| `sign` | :white_check_mark: | white_check_mark | |
| `sin` | :white_check_mark: | :white_check_mark: | |
| `sinh` | :x: | :x: | |
| `sort` | :x: | :x: | |
Expand All @@ -116,8 +116,8 @@
| `squeeze` | :white_check_mark: | :white_check_mark: | |
| `sub` | :white_check_mark: | :white_check_mark: | |
| `tan` | :white_check_mark: | :white_check_mark: | |
| `tie_in` | :x: | :x: | Deprecated in JAX |
| `top_k` | :x: | :x: | |
| `tie_in` | :x: | n/a | Deprecated in JAX |
| `top_k` | :x: | n/a | |
| `transpose` | :white_check_mark: | :x: | Copies the input tensor |
| `zeta` | :x: | :x: | |

Expand Down

0 comments on commit e418dac

Please sign in to comment.