UOp(Ops.LINEAR, arg=None, src=(
  UOp(Ops.CALL, arg=CallInfo(None, None, False, False), src=(
    UOp(Ops.SINK, arg=KernelInfo(name='test', applied_opts=(), opts_to_apply=None, estimates=None, beam=0), src=(
      UOp(Ops.END, arg=None, src=(
        UOp(Ops.STORE, arg=None, src=(
          UOp(Ops.INDEX, arg=None, src=(
            UOp(Ops.PARAM, arg=ParamArg(0, dtypes.float, 5, device='CPU'), src=()),
            x6:=UOp(Ops.RANGE, arg=(1, AxisType.WEAK), src=(
              UOp(Ops.CONST, arg=5, src=()),)),)),
          UOp(Ops.REDUCE, arg=(Ops.ADD, 0), src=(
            UOp(Ops.ADD, arg=None, src=(
              UOp(Ops.WHERE, arg=None, src=(
                x11:=UOp(Ops.AND, arg=None, src=(
                  UOp(Ops.AND, arg=None, src=(
                    UOp(Ops.CMPNE, arg=None, src=(
                      UOp(Ops.CMPLT, arg=None, src=(
                        x6,
                        x15:=UOp(Ops.CONST, arg=1, src=()),)),
                      x16:=UOp(Ops.CONST, arg=True, src=()),)),
                    UOp(Ops.CMPLT, arg=None, src=(
                      x6,
                      x18:=UOp(Ops.CONST, arg=4, src=()),)),)),
                  UOp(Ops.AND, arg=None, src=(
                    UOp(Ops.CMPNE, arg=None, src=(
                      UOp(Ops.CMPLT, arg=None, src=(
                        x22:=UOp(Ops.RANGE, arg=(0, AxisType.REDUCE), src=(
                          x18,)),
                        x15,)),
                      x16,)),
                    UOp(Ops.CMPLT, arg=None, src=(
                      x22,
                      x24:=UOp(Ops.CONST, arg=3, src=()),)),)),)),
                UOp(Ops.INDEX, arg=None, src=(
                  UOp(Ops.PARAM, arg=ParamArg(1, dtypes.float, 6, device='CPU'), src=()),
                  UOp(Ops.WHERE, arg=None, src=(
                    x11,
                    UOp(Ops.ADD, arg=None, src=(
                      UOp(Ops.ADD, arg=None, src=(
                        UOp(Ops.MUL, arg=None, src=(
                          x22,
                          x24,)),
                        x6,)),
                      UOp(Ops.CONST, arg=-4, src=()),)),
                    UOp(Ops.CONST, arg=Invalid, src=()),)),)),
                UOp(Ops.CONST, arg=ConstFloat(0.0), src=()),)),
              UOp(Ops.CONST, arg=ConstFloat(10.0), src=()),)),
            x22,)),)),
        x6,)),)),
    UOp(Ops.BUFFER, arg=ParamArg(2, dtypes.float, 5, device='CPU'), src=()),
    UOp(Ops.BUFFER, arg=ParamArg(1, dtypes.float, 6, device='CPU'), src=()),)),))