EnzymeAD/Reactant.jl

Tracking issue for shortcuts to MLIR ops

Open

#273 opened on Nov 14, 2024

View on GitHub
 (0 comments) (0 reactions) (0 assignees)Julia (64 forks)github user discovery
good first issuehelp wanted

Repository metrics

Stars
 (336 stars)
PR merge metrics
 (PR metrics pending)

Description

  • stablehlo

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

    • acos
    • acosh
    • _asin_acos_kernel --> should not be used directly
    • asin
    • asinh
    • atan
    • atanh
    • 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
    • conj
    • constant_like
    • constant
    • cosh
    • digamma
    • erf_inv
    • erf
    • erfc
    • is_inf
    • is_neg_inf
    • is_pos_inf
    • lgamma
    • next_after
    • polygamma
    • sinh
    • tan
    • top_k
    • zeta
  • enzyme

    • addTo
    • autodiff
    • batch
    • fwddiff
    • genericAdjoint
    • get
    • init
    • placeholder
    • pop
    • push
    • set

Contributor guide