Skip to content

Commit

Permalink
Remove comments saying dynamic result means speculatable
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael Levesque-Dion committed May 1, 2024
1 parent 7a19019 commit 8fec76b
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions stablehlo/dialect/Base.h
Original file line number Diff line number Diff line change
Expand Up @@ -257,8 +257,8 @@ void writeEnumAttribute(EnumTypeAttr val, DialectBytecodeWriter &writer) {

// Determines the speculatability for a shaped operation `op` with `shapeCount`
// shape operands. The last `count` operands are assumed to be shape operands.
// To be speculatable, such an op must either have a fully dynamic result type
// or have only static inputs and constant shape operands.
// To be speculatable, such an op must have only static inputs and constant
// shape operands.
mlir::Speculation::Speculatability getShapedSpeculatability(Operation *op,
int64_t shapeCount);

Expand Down
4 changes: 2 additions & 2 deletions stablehlo/dialect/Base.td
Original file line number Diff line number Diff line change
Expand Up @@ -372,8 +372,8 @@ def HLO_SpeculatableIfAllInputsStaticAndShapeConstantImplTrait
: HLO_NativeOpTrait<"SpeculatableIfAllInputsStaticAndShapeConstantImplTrait">;

// This trait is the same as HLO_SpeculatableIfAllInputsStatic, but for ops that
// take a shape as their last operand. Such ops are speculatable if either the
// output is dynamic or all inputs are static and the shape is constant.
// take a shape as their last operand. Such ops are speculatable if all inputs
// are static and the shape is constant.
def HLO_SpeculatableIfAllInputsStaticAndShapeConstant : TraitList<[
ConditionallySpeculatable, HLO_SpeculatableIfAllInputsStaticAndShapeConstantImplTrait]>;

Expand Down

0 comments on commit 8fec76b

Please sign in to comment.