EnzymeAD/Reactant.jl

Tracking issue for shortcuts to MLIR ops

Ouverte

#273 ouverte le 14 nov. 2024

 (0 commentaire) (0 réaction) (0 personne assignée)Julia (72 forks)github user discovery
good first issuehelp wanted

Métriques du dépôt

Stars
 (365 étoiles)
Métriques de merge PR
 (Merge moyen 2j 10h) (36 PRs mergées en 30 j)

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