EnzymeAD/Enzyme-JAX

Tracking issue for missing HLO derivatives

Ouverte

#88 ouverte le 1 juin 2024

 (4 commentaires) (0 réaction) (1 personne assignée)MLIR (45 forks)auto 404
good first issuehelp wanted

Métriques du dépôt

Stars
 (126 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

Description

NOTE: Strikethrough ops are deliberately not annotated.

  • StableHLO
    • AbsOp
    • AddOp
    • AfterAllOp
    • AllGatherOp
    • AllReduceOp
    • AllToAllOp
    • AndOp
    • Atan2Op #90
    • BatchNormGradOp
    • BatchNormInferenceOp
    • BatchNormTrainingOp
    • BitcastConvertOp
    • BroadcastInDimOp
    • CaseOp
    • CbrtOp #90
    • CeilOp #90
    • CholeskyOp
    • ClampOp #148
    • CollectiveBroadcastOp
    • CollectivePermuteOp
    • CompareOp
    • ComplexOp #90
    • CompositeOp
    • ConcatenateOp
    • ConstantOp
    • ConvertOp
    • ConvolutionOp #178
    • CosineOp
    • ClzOp
    • CustomCallOp
    • DivOp
    • DotGeneralOp
    • DynamicBroadcastInDimOp
    • DynamicConvOp
    • DynamicGatherOp
    • DynamicIotaOp
    • DynamicPadOp
    • DynamicReshapeOp
    • DynamicSliceOp #220
    • DynamicUpdateSliceOp #159
    • ExpOp
    • Expm1Op #90
    • FftOp (current rule is wrong!) #90
    • FloorOp #90
    • GatherOp
    • GetDimensionSizeOp
    • GetTupleElementOp
    • IfOp #149
    • ImagOp #145
    • InfeedOp
    • IotaOp #90
    • IsFiniteOp #90
    • LogOp
    • Logp1Op #90
    • LogisticOp #90
    • MapOp
    • MaxOp
    • MinOp #90
    • MulOp
    • NegateOp
    • NotOp
    • OptimizationBarrierOp
    • OrOp
    • OutfeedOp
    • PadOp
    • PartitionIdOp
    • PopcntOp
    • PowOp
    • RealOp #145
    • RecvOp
    • ReduceOp
    • ReducePrecisionOp
    • ReduceScatterOp
    • ReduceWindowOp
    • RemainderOp
    • ReplicaIdOp
    • ReshapeOp
    • ReverseOp #90
    • RngOp #90
    • RngBitGeneratorOp #90
    • RoundOp #90
    • RoundNearestEvenOp #90
    • RsqrtOp
    • ScatterOp
    • SelectOp
    • SelectAndScatterOp
    • SendOp
    • ShiftLeftOp
    • ShiftRightArithmeticOp
    • ShiftRightLogicalOp
    • SignOp #90
    • SineOp
    • SliceOp
    • SortOp
    • SqrtOp
    • SubtractOp
    • TanhOp
    • TransposeOp
    • TriangularSolveOp
    • #157
    • UniformDequantizeOp
    • UniformQuantizeOp
    • WhileOp #160
    • XorOp
    • Deprecated operations in StableHLO
      • BroadcastOp
      • CreateTokenOp
      • CrossReplicaSumOp
      • DotOp
      • EinsumOp
      • TorchIndexSelectOp
      • UnaryEinsumOp
  • CHLO
    • Binary Element-wise Operations
      • BroadcastAddOp
      • BroadcastAtan2Op
      • BroadcastDivOp
      • BroadcastMaxOp
      • BroadcastMinOp
      • BroadcastMulOp
      • BroadcastNextAfterOp
      • BroadcastPolygammaOp
      • BroadcastPowOp
      • BroadcastRemOp
      • BroadcastShiftLeftOp
      • BroadcastShiftRightArithmeticOp
      • BroadcastShiftRightLogicalOp
      • BroadcastSubOp
      • BroadcastZetaOp
    • Binary Logical Element-wise Operations
      • BroadcastAndOp
      • BroadcastOrOp
      • BroadcastXorOp
    • Non-broadcasting Binary Operations
      • NextAfterOp
      • PolygammaOp #90
      • ZetaOp
    • ComplexOp
    • Unary Element-wise Operations
      • AcosOp #90
      • AcoshOp #90
      • AsinOp #90
      • AsinhOp #90
      • AtanOp #90
      • AtanhOp #90
      • BesselI1eOp
      • ConjOp #90
      • CoshOp #90
      • SinhOp #90
      • TanOp #90
      • ConstantOp (shared with StableHLO_ConstantOp)
      • ConstantLikeOp
      • DigammaOp #90
      • ErfOp
      • ErfInvOp
      • ErfcOp
      • IsInfOp #90
      • IsNegInfOp #90
      • IsPosInfOp #90
      • LgammaOp
    • BroadcastCompareOp
    • BroadcastSelectOp
    • TopKOp

Guide contributeur