EnzymeAD / EnzymeAD/Enzyme-JAX
Tracking issue for missing Batch Op Interface
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
NOTE: Strikethrough ops are deliberately not implemented because the default broadcasting behavior of enzyme batch is enough.
- [ ] StableHLO
- [x] ~AbsOp~
- [x] ~AddOp~
- [x] ~AfterAllOp~
- [ ] AllGatherOp
- [ ] AllReduceOp
- [ ] AllToAllOp
- [x] ~AndOp~
- [x] ~Atan2Op~
- [ ] BatchNormGradOp
- [ ] BatchNormInferenceOp
- [ ] BatchNormTrainingOp
- [x] ~BitcastConvertOp~
- [x] BroadcastInDimOp
- [ ] CaseOp
- [x] ~CbrtOp~
- [x] ~CeilOp~
- [x] ~CholeskyOp~
- [x] ~ClampOp~
- [ ] CollectiveBroadcastOp
- [ ] CollectivePermuteOp
- [x] ~CompareOp~
- [x] ~ComplexOp~
- [ ] CompositeOp
- [x] ConcatenateOp
- [x] ConstantOp
- [x] ~ConvertOp~
- [x] ConvolutionOp #151
- [x] ~CosineOp~
- [x] ~ClzOp~
- [x] CustomCallOp
- [x] ~DivOp~
- [x] DotGeneralOp
- [ ] DynamicBroadcastInDimOp
- [ ] DynamicConvOp
- [ ] DynamicGatherOp
- [ ] DynamicIotaOp
- [ ] DynamicPadOp
- [ ] DynamicReshapeOp
- [x] DynamicSliceOp
- [x] DynamicUpdateSliceOp
- [x] ~ExpOp~
- [x] ~Expm1Op~
- [x] ~FftOp~
- [x] ~FloorOp~
- [x] GatherOp
- [x] GetDimensionSizeOp
- [x] IfOp #151
- [x] ~ImagOp~
- [ ] InfeedOp
- [x] IotaOp
- [x] ~IsFiniteOp~
- [x] ~LogOp~
- [x] ~Logp1Op~
- [x] ~LogisticOp~
- [x] ~MaxOp~
- [x] ~MinOp~
- [x] ~MulOp~
- [x] ~NegateOp~
- [x] ~NotOp~
- [ ] OptimizationBarrierOp
- [x] ~OrOp~
- [ ] OutfeedOp
- [ ] PadOp
- [ ] PartitionIdOp
- [x] ~PopcntOp~
- [x] ~PowOp~
- [x] ~RealOp~
- [ ] RecvOp
- [x] ReduceOp
- [x] ~ReducePrecisionOp~
- [ ] ReduceScatterOp
- [x] ReduceWindowOp
- [x] ~RemainderOp~
- [ ] ReplicaIdOp
- [x] ReshapeOp
- [x] ReverseOp #151
- [ ] RngBitGeneratorOp
- [x] ~RoundOp~
- [x] ~RoundNearestEvenOp~
- [x] ~RsqrtOp~
- [ ] ScatterOp
- [x] SelectOp
- [ ] SelectAndScatterOp
- [ ] SendOp
- [x] ~ShiftLeftOp~
- [x] ~ShiftRightArithmeticOp~
- [x] ~ShiftRightLogicalOp~
- [x] ~SignOp~
- [x] ~SineOp~
- [x] SliceOp
- [x] SortOp
- [x] ~SqrtOp~
- [x] ~SubtractOp~
- [x] ~TanhOp~
- [x] TransposeOp
- [x] ~TriangularSolveOp~
- [ ] UniformDequantizeOp
- [ ] UniformQuantizeOp
- [x] WhileOp #151
- [x] ~XorOp~
- [ ] Deprecated operations in StableHLO
- [ ] RngOp
- [ ] GetTupleElementOp
- [ ] BroadcastOp
- [ ] CreateTokenOp
- [ ] CrossReplicaSumOp
- [ ] DotOp
- [ ] EinsumOp
- [x] ~TorchIndexSelectOp~
- [ ] UnaryEinsumOp
- [ ] TupleOp
- [ ] MapOp
- [ ] CHLO
- [x] Binary Element-wise Operations
- [x] ~BroadcastAddOp~
- [x] ~BroadcastAtan2Op~
- [x] ~BroadcastDivOp~
- [x] ~BroadcastMaxOp~
- [x] ~BroadcastMinOp~
- [x] ~BroadcastMulOp~
- [x] ~BroadcastNextAfterOp~
- [x] ~BroadcastPolygammaOp~
- [x] ~BroadcastPowOp~
- [x] ~BroadcastRemOp~
- [x] ~BroadcastShiftLeftOp~
- [x] ~BroadcastShiftRightArithmeticOp~
- [x] ~BroadcastShiftRightLogicalOp~
- [x] ~BroadcastSubOp~
- [x] ~BroadcastZetaOp~
- [x] Binary Logical Element-wise Operations
- [x] ~BroadcastAndOp~
- [x] ~BroadcastOrOp~
- [x] ~BroadcastXorOp~
- [ ] Non-broadcasting Binary Operations
- [ ] NextAfterOp
- [ ] PolygammaOp
- [ ] ZetaOp
- [ ] `ComplexOp`
- [x] Unary Element-wise Operations
- [x] ~AcosOp~
- [x] ~AcoshOp~
- [x] ~AsinOp~
- [x] ~AsinhOp~
- [x] ~AtanOp~
- [x] ~AtanhOp~
- [x] ~BesselI1eOp~
- [x] ~ConjOp~
- [x] ~CoshOp~
- [x] ~SinhOp~
- [x] ~TanOp~
- [x] ~ConstantOp~ (shared with `StableHLO_ConstantOp`)
- [x] ~ConstantLikeOp~
- [x] ~DigammaOp~
- [x] ~ErfOp~
- [x] ~ErfInvOp~
- [x] ~ErfcOp~
- [x] ~IsInfOp~
- [x] ~IsNegInfOp~
- [x] ~IsPosInfOp~
- [ ] LgammaOp
- [ ] `BroadcastCompareOp`
- [ ] `BroadcastSelectOp`
- [x] ~`TopKOp`~
- [ ] EnzymeXLA
- [ ] KernelCallOp
- [ ] JitCallOp
- [ ] GetStreamOp
- [ ] Memref2PointerOp
- [ ] Pointer2MemrefOp
- [ ] AffineScopeOp
- [ ] RotateOp
- [ ] WrapOp
- [ ] ExtendOp
- [x] ~CommRegionOp~
- [x] ~LUFactorizationOp~
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.