Extending Select to support Int32

glow
1

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?

2

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