EnzymeAD/Enzyme-JAX

Tracking issue for missing Batch Op Interface

Ouverte

#152 ouverte le 5 nov. 2024

Voir sur GitHub
 (5 commentaires) (0 réaction) (0 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 implemented because the default broadcasting behavior of enzyme batch is enough.

  • StableHLO
    • AbsOp
    • AddOp
    • AfterAllOp
    • AllGatherOp
    • AllReduceOp
    • AllToAllOp
    • AndOp
    • Atan2Op
    • BatchNormGradOp
    • BatchNormInferenceOp
    • BatchNormTrainingOp
    • BitcastConvertOp
    • BroadcastInDimOp
    • CaseOp
    • CbrtOp
    • CeilOp
    • CholeskyOp
    • ClampOp
    • CollectiveBroadcastOp
    • CollectivePermuteOp
    • CompareOp
    • ComplexOp
    • CompositeOp
    • ConcatenateOp
    • ConstantOp
    • ConvertOp
    • ConvolutionOp #151
    • CosineOp
    • ClzOp
    • CustomCallOp
    • DivOp
    • DotGeneralOp
    • DynamicBroadcastInDimOp
    • DynamicConvOp
    • DynamicGatherOp
    • DynamicIotaOp
    • DynamicPadOp
    • DynamicReshapeOp
    • DynamicSliceOp
    • DynamicUpdateSliceOp
    • ExpOp
    • Expm1Op
    • FftOp
    • FloorOp
    • GatherOp
    • GetDimensionSizeOp
    • IfOp #151
    • ImagOp
    • InfeedOp
    • IotaOp
    • IsFiniteOp
    • LogOp
    • Logp1Op
    • LogisticOp
    • MaxOp
    • MinOp
    • MulOp
    • NegateOp
    • NotOp
    • OptimizationBarrierOp
    • OrOp
    • OutfeedOp
    • PadOp
    • PartitionIdOp
    • PopcntOp
    • PowOp
    • RealOp
    • RecvOp
    • ReduceOp
    • ReducePrecisionOp
    • ReduceScatterOp
    • ReduceWindowOp
    • RemainderOp
    • ReplicaIdOp
    • ReshapeOp
    • ReverseOp #151
    • RngBitGeneratorOp
    • RoundOp
    • RoundNearestEvenOp
    • RsqrtOp
    • ScatterOp
    • SelectOp
    • SelectAndScatterOp
    • SendOp
    • ShiftLeftOp
    • ShiftRightArithmeticOp
    • ShiftRightLogicalOp
    • SignOp
    • SineOp
    • SliceOp
    • SortOp
    • SqrtOp
    • SubtractOp
    • TanhOp
    • TransposeOp
    • TriangularSolveOp
    • UniformDequantizeOp
    • UniformQuantizeOp
    • WhileOp #151
    • XorOp
    • Deprecated operations in StableHLO
      • RngOp
      • GetTupleElementOp
      • BroadcastOp
      • CreateTokenOp
      • CrossReplicaSumOp
      • DotOp
      • EinsumOp
      • TorchIndexSelectOp
      • UnaryEinsumOp
      • TupleOp
      • MapOp
  • 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
      • ZetaOp
    • ComplexOp
    • Unary Element-wise Operations
      • AcosOp
      • AcoshOp
      • AsinOp
      • AsinhOp
      • AtanOp
      • AtanhOp
      • BesselI1eOp
      • ConjOp
      • CoshOp
      • SinhOp
      • TanOp
      • ConstantOp (shared with StableHLO_ConstantOp)
      • ConstantLikeOp
      • DigammaOp
      • ErfOp
      • ErfInvOp
      • ErfcOp
      • IsInfOp
      • IsNegInfOp
      • IsPosInfOp
      • LgammaOp
    • BroadcastCompareOp
    • BroadcastSelectOp
    • TopKOp
  • EnzymeXLA
    • KernelCallOp
    • JitCallOp
    • GetStreamOp
    • Memref2PointerOp
    • Pointer2MemrefOp
    • AffineScopeOp
    • RotateOp
    • WrapOp
    • ExtendOp
    • CommRegionOp
    • LUFactorizationOp

Guide contributeur

Direction de recherche
Étudiez l'implémentation existante de l'interface d'opérations par lots dans Enzyme et implémentez les opérations par lots manquantes pour les opérations répertoriées en suivant le modèle établi. Commencez par une opération simple comme ConstantOp ou IotaOp pour comprendre le modèle, puis passez à des opérations plus complexes comme ConvolutionOp.
Stack technique
Aucun
Domaine
compilersmachine learning
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Plus d'une semaine
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
MLIRC++
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
25