It is not in the list of isOpSupported.

It is supported, perhaps you just missed it, it’s listed here:

  

      github.com
  

  
    pytorch/glow/blob/778d44eff16f8dfac25503e0f917531d71bc950b/lib/Backends/OpenCL/OpenCL.cpp#L1931



      
      
                  {ElemKind::FloatTy}, {},
                  {ConvolutionGradNode::GradOfInputNamedInputIdx});
          
          
  case Kinded::Kind::SoftMaxGradNodeKind:
              return NI.allInputsAndOutputsHaveSameElemKind(
                         {ElemKind::FloatTy}, {SoftMaxGradNode::SelectedIdx},
                         {SoftMaxGradNode::GradOfInputNamedSelectedIdx}) &&
                     (NI.getInElemTy(SoftMaxGradNode::SelectedIdx) == ElemKind::Int64ITy);
          
          
  case Kinded::Kind::SaveNodeKind:
            case Kinded::Kind::ReshapeNodeKind:
            case Kinded::Kind::OCLBatchedReduceAddNodeKind:
            case Kinded::Kind::TraceEventNodeKind:
              // These work regardless of the underlying type.
              return true;
          
          
  default:
              return false;
            }
          }