EnzymeAD / EnzymeAD/Reactant.jl

Tracking issue for shortcuts to MLIR ops

オープン
#273 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る
good first issue help wanted
主要言語
Julia
スター
370
フォーク
74
平均マージ
18時間 47分
マージ済み PR(30日)
30

説明

- [ ] `stablehlo`
- [x] `abs`
- [x] `add`
- [x] `after_all`
- [ ] `all_gather`
- [ ] `all_reduce`
- [ ] `all_to_all`
- [x] `and`
- [x] `atan2`
- [x] `batch_norm_grad`
- [x] `batch_norm_inference`
- [x] `batch_norm_training`
- [ ] `bitcast_convert`
- [x] `broadcast_in_dim`
- ~`broadcast`~ -> on deprecation process
- [x] `case`
- [x] `cbrt`
- [x] `ceil`
- [x] `cholesky`
- [x] `clamp`
- [x] `count_leading_zeros`
- [ ] `collective_broadcast`
- [ ] `collective_permute`
- [x] `compare`
- [x] `complex`
- [ ] `composite`
- [x] `concatenate`
- [x] `constant`
- [x] `convert` #347
- [ ] `convolution` #428
- [x] `cosine`
- ~`create_token`~ -> on deprecation process
- ~`cross_replica_sum`~ -> on deprecation process
- [ ] `custom_call`
- [x] `divide`
- [x] `dot_general` #330
- ~`dot`~ -> on deprecation process
- [ ] `dynamic_broadcast_in_dim`
- [ ] `dynamic_conv`
- [ ] `dynamic_gather`
- [ ] `dynamic_iota`
- [ ] `dynamic_pad`
- [ ] `dynamic_reshape`
- [x] `dynamic_slice`
- [x] `dynamic_update_slice`
- [x] einsum -> on deprecation process, might be moved to CHLO
- [x] `exponential`
- [x] `exponential_minus_one`
- [x] `fft`
- [x] `floor`
- [x] `gather`
- [x] `get_dimension_size`
- ~`get_tuple_element`~ -> on deprecation process
- [x] `if_`
- [x] `imag`
- [ ] `infeed`
- [x] `iota`
- [x] `is_finite`
- [x] `log_plus_one`
- [x] `log`
- [x] `logistic`
- ~`map`~ -> on deprecation process
- [x] `maximum`
- [x] `minimum`
- [x] `multiply`
- [x] `negate`
- [x] `not`
- [x] `optimization_barrier`
- [x] `or`
- [x] `outfeed`
- [x] `pad`
- [x] `partition_id`
- [x] `popcnt`
- [x] `power`
- ~`real_dynamic_slice`~ -> on deprecation process
- [x] `real`
- [x] `recv`
- [x] `reduce`
- [ ] `reduce_precision`
- [ ] `reduce_scatter`
- [ ] `reduce_window`
- [x] `remainder`
- [x] `replica_id`
- [x] `reshape`
- [x] `reverse`
- [x] `rng_bit_generator`
- ~`rng`~ -> on deprecation process
- [x] `round_nearest_even`
- [x] `round_nearest_afz`
- [x] `rsqrt`
- [x] `scatter`
- [ ] `select_and_scatter`
- [x] `select` #328
- [x] `send`
- [x] `set_dimension_size`
- [x] `shift_left`
- [x] `shift_right_arithmetic`
- [x] `shift_right_logical`
- [x] `sign`
- [x] `sine`
- [x] `slice`
- [x] `sort`
- [x] `sqrt`
- [x] `subtract`
- [x] `tan`
- [x] `tanh`
- ~`torch_index_select`~ -> on deprecation process
- [x] `transpose`
- [x] `triangular_solve`
- ~`tuple`~ -> on deprecation process
- ~`unary_einsum`~ -> on deprecation process
- [ ] `uniform_dequantize`
- [ ] `uniform_quantize`
- [x] `while_`
- [x] `xor`

- [ ] `chlo`
- [x] `acos`
- [x] `acosh`
- ~`_asin_acos_kernel`~ --> should not be used `directly`
- [x] `asin`
- [x] `asinh`
- [x] `atan`
- [x] `atanh`
- [x] `bessel_i1e`
- [ ] `broadcast_add`
- [ ] `broadcast_and`
- [ ] `broadcast_atan2`
- [ ] `broadcast_compare`
- [ ] `broadcast_complex`
- [ ] `broadcast_divide`
- [ ] `broadcast_maximum`
- [ ] `broadcast_minimum`
- [ ] `broadcast_multiply`
- [ ] `broadcast_next_after`
- [ ] `broadcast_or`
- [ ] `broadcast_polygamma`
- [ ] `broadcast_power`
- [ ] `broadcast_remainder`
- [ ] `broadcast_select`
- [ ] `broadcast_shift_left`
- [ ] `broadcast_shift_right_arithmetic`
- [ ] `broadcast_shift_right_logical`
- [ ] `broadcast_subtract`
- [ ] `broadcast_xor`
- [ ] `broadcast_zeta`
- [x] `conj`
- [ ] `constant_like`
- [ ] `constant`
- [x] `cosh`
- [x] `digamma`
- [x] `erf_inv`
- [x] `erf`
- [x] `erfc`
- [x] `is_inf`
- [x] `is_neg_inf`
- [x] `is_pos_inf`
- [x] `lgamma`
- [x] `next_after`
- [x] `polygamma`
- [x] `sinh`
- [x] `tan`
- [x] `top_k`
- [x] `zeta`

- [ ] `enzyme`
- [ ] `addTo`
- [ ] `autodiff`
- [x] `batch`
- [ ] `fwddiff`
- [ ] `genericAdjoint`
- [ ] `get`
- [ ] `init`
- [ ] `placeholder`
- [ ] `pop`
- [ ] `push`
- [ ] `set`

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。