From e418dac12791971659fdfd562700d2925e2f7b7a Mon Sep 17 00:00:00 2001 From: Paul Balanca Date: Wed, 27 Sep 2023 21:57:31 +0100 Subject: [PATCH] Update JAX LAX compatibility table. (#36) --- docs/operations.md | 96 +++++++++++++++++++++++----------------------- 1 file changed, 48 insertions(+), 48 deletions(-) diff --git a/docs/operations.md b/docs/operations.md index b009d2d..3b040e4 100644 --- a/docs/operations.md +++ b/docs/operations.md @@ -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: | | @@ -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: | | @@ -47,33 +47,33 @@ | `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: | | @@ -81,32 +81,32 @@ | `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: | | @@ -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: | |