Extending Select to support Int32

How can one go about extending Select to work on tensors with element type int32, specifically in the interpreter backend? Or, is it not recommended to extend a single op on a single backend?

Hi @alannnna, Totally reasonable to extend this. I don’t think we have a doc here on what to do, but off the top of my head it should be as simple as:

  • Update Interpreter::isOpSupported() to include ElemKind::Int32ITy
  • Update BoundInterpreterFunction::fwdElementExpInst() to support it
  • (probably not necessary) May need to update SelectNode::verify() if there is any verification that fails
1 Like