EnzymeAD/Reactant.jl

Tracking issue for shortcuts to MLIR ops

Open

#273 ouverte le 14 nov. 2024

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)Julia (64 forks)github user discovery
good first issuehelp wanted

Métriques du dépôt

Stars
 (336 stars)
Métriques de merge PR
 (Métriques PR en attente)

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

Guide contributeur