HloModule jit_lu_pivots_to_permutation.1466

%fused_computation (param_0: s32[75], param_1.733: s32[75], param_2.835: s32[], param_3.878: s32[75]) -> s32[75] {
  %param_0 = s32[75]{0} parameter(0)
  %constant_4373 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.733 = s32[75]{0} parameter(1)
  %param_2.835 = s32[] parameter(2)
  %constant_5240 = s32[] constant(15)
  %add.1482 = s32[] add(s32[] %param_2.835, s32[] %constant_5240), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1090 = pred[] compare(s32[] %add.1482, s32[] %constant_4373), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5238 = s32[] constant(90)
  %add.1481 = s32[] add(s32[] %param_2.835, s32[] %constant_5238), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.602 = s32[] select(pred[] %compare.1090, s32[] %add.1481, s32[] %add.1482), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.699 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.733, s32[] %select.602), dynamic_slice_sizes={1}
  %bitcast.225 = s32[] bitcast(s32[1]{0} %dynamic-slice.699), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1089 = pred[] compare(s32[] %bitcast.225, s32[] %constant_4373), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5236 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1480 = s32[] add(s32[] %bitcast.225, s32[] %constant_5236), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.601 = s32[] select(pred[] %compare.1089, s32[] %add.1480, s32[] %bitcast.225), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.664 = pred[] compare(s32[] %constant_4373, s32[] %select.601), direction=LE
  %constant_4372 = s32[] constant(74)
  %compare.663 = pred[] compare(s32[] %constant_4372, s32[] %select.601), direction=GE
  %and.248 = pred[] and(pred[] %compare.664, pred[] %compare.663)
  %bitcast.81 = pred[1]{0} bitcast(pred[] %and.248)
  %param_3.878 = s32[75]{0} parameter(3)
  %dynamic-slice.818 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.878, s32[] %select.602), dynamic_slice_sizes={1}
  %dynamic-slice.539 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0, s32[] %select.601), dynamic_slice_sizes={1}
  %select.229 = s32[1]{0} select(pred[1]{0} %bitcast.81, s32[1]{0} %dynamic-slice.818, s32[1]{0} %dynamic-slice.539)
  ROOT %dynamic-update-slice.221 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0, s32[1]{0} %select.229, s32[] %select.601), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.1 (param_0.1: s32[75], param_1.731: s32[], param_2.834: s32[75]) -> s32[75] {
  %param_0.1 = s32[75]{0} parameter(0)
  %constant_4375 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.731 = s32[] parameter(1)
  %constant_5218 = s32[] constant(15)
  %add.1470 = s32[] add(s32[] %param_1.731, s32[] %constant_5218), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1081 = pred[] compare(s32[] %add.1470, s32[] %constant_4375), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5216 = s32[] constant(90)
  %add.1469 = s32[] add(s32[] %param_1.731, s32[] %constant_5216), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.593 = s32[] select(pred[] %compare.1081, s32[] %add.1469, s32[] %add.1470), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.666 = pred[] compare(s32[] %constant_4375, s32[] %select.593), direction=LE
  %constant_4374 = s32[] constant(74)
  %compare.665 = pred[] compare(s32[] %constant_4374, s32[] %select.593), direction=GE
  %and.249 = pred[] and(pred[] %compare.666, pred[] %compare.665)
  %bitcast.82 = pred[1]{0} bitcast(pred[] %and.249)
  %param_2.834 = s32[75]{0} parameter(2)
  %dynamic-slice.697 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.834, s32[] %select.593), dynamic_slice_sizes={1}
  %bitcast.223 = s32[] bitcast(s32[1]{0} %dynamic-slice.697), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1085 = pred[] compare(s32[] %bitcast.223, s32[] %constant_4375), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5225 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1474 = s32[] add(s32[] %bitcast.223, s32[] %constant_5225), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.596 = s32[] select(pred[] %compare.1085, s32[] %add.1474, s32[] %bitcast.223), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.540 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.1, s32[] %select.596), dynamic_slice_sizes={1}
  %dynamic-slice.816 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.1, s32[] %select.593), dynamic_slice_sizes={1}
  %select.230 = s32[1]{0} select(pred[1]{0} %bitcast.82, s32[1]{0} %dynamic-slice.540, s32[1]{0} %dynamic-slice.816)
  ROOT %dynamic-update-slice.222 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.1, s32[1]{0} %select.230, s32[] %select.593), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.4 (param_0.8: s32[75], param_1.729: s32[75], param_2.833: s32[], param_3.873: s32[75]) -> s32[75] {
  %param_0.8 = s32[75]{0} parameter(0)
  %constant_4380 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.729 = s32[75]{0} parameter(1)
  %param_2.833 = s32[] parameter(2)
  %constant_5197 = s32[] constant(14)
  %add.1457 = s32[] add(s32[] %param_2.833, s32[] %constant_5197), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1075 = pred[] compare(s32[] %add.1457, s32[] %constant_4380), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5195 = s32[] constant(89)
  %add.1456 = s32[] add(s32[] %param_2.833, s32[] %constant_5195), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.587 = s32[] select(pred[] %compare.1075, s32[] %add.1456, s32[] %add.1457), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.694 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.729, s32[] %select.587), dynamic_slice_sizes={1}
  %bitcast.221 = s32[] bitcast(s32[1]{0} %dynamic-slice.694), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1074 = pred[] compare(s32[] %bitcast.221, s32[] %constant_4380), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5193 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1455 = s32[] add(s32[] %bitcast.221, s32[] %constant_5193), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.585 = s32[] select(pred[] %compare.1074, s32[] %add.1455, s32[] %bitcast.221), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.670 = pred[] compare(s32[] %constant_4380, s32[] %select.585), direction=LE
  %constant_4379 = s32[] constant(74)
  %compare.669 = pred[] compare(s32[] %constant_4379, s32[] %select.585), direction=GE
  %and.250 = pred[] and(pred[] %compare.670, pred[] %compare.669)
  %bitcast.84 = pred[1]{0} bitcast(pred[] %and.250)
  %param_3.873 = s32[75]{0} parameter(3)
  %dynamic-slice.814 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.873, s32[] %select.587), dynamic_slice_sizes={1}
  %dynamic-slice.542 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.8, s32[] %select.585), dynamic_slice_sizes={1}
  %select.233 = s32[1]{0} select(pred[1]{0} %bitcast.84, s32[1]{0} %dynamic-slice.814, s32[1]{0} %dynamic-slice.542)
  ROOT %dynamic-update-slice.223 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.8, s32[1]{0} %select.233, s32[] %select.585), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.5 (param_0.9: s32[75], param_1.727: s32[], param_2.832: s32[75]) -> s32[75] {
  %param_0.9 = s32[75]{0} parameter(0)
  %constant_4382 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.727 = s32[] parameter(1)
  %constant_5175 = s32[] constant(14)
  %add.1444 = s32[] add(s32[] %param_1.727, s32[] %constant_5175), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1066 = pred[] compare(s32[] %add.1444, s32[] %constant_4382), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5173 = s32[] constant(89)
  %add.1443 = s32[] add(s32[] %param_1.727, s32[] %constant_5173), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.578 = s32[] select(pred[] %compare.1066, s32[] %add.1443, s32[] %add.1444), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.673 = pred[] compare(s32[] %constant_4382, s32[] %select.578), direction=LE
  %constant_4381 = s32[] constant(74)
  %compare.672 = pred[] compare(s32[] %constant_4381, s32[] %select.578), direction=GE
  %and.251 = pred[] and(pred[] %compare.673, pred[] %compare.672)
  %bitcast.85 = pred[1]{0} bitcast(pred[] %and.251)
  %param_2.832 = s32[75]{0} parameter(2)
  %dynamic-slice.692 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.832, s32[] %select.578), dynamic_slice_sizes={1}
  %bitcast.219 = s32[] bitcast(s32[1]{0} %dynamic-slice.692), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1069 = pred[] compare(s32[] %bitcast.219, s32[] %constant_4382), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5182 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1448 = s32[] add(s32[] %bitcast.219, s32[] %constant_5182), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.581 = s32[] select(pred[] %compare.1069, s32[] %add.1448, s32[] %bitcast.219), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.543 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.9, s32[] %select.581), dynamic_slice_sizes={1}
  %dynamic-slice.812 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.9, s32[] %select.578), dynamic_slice_sizes={1}
  %select.235 = s32[1]{0} select(pred[1]{0} %bitcast.85, s32[1]{0} %dynamic-slice.543, s32[1]{0} %dynamic-slice.812)
  ROOT %dynamic-update-slice.224 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.9, s32[1]{0} %select.235, s32[] %select.578), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.8 (param_0.16: s32[75], param_1.725: s32[75], param_2.831: s32[], param_3.868: s32[75]) -> s32[75] {
  %param_0.16 = s32[75]{0} parameter(0)
  %constant_4387 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.725 = s32[75]{0} parameter(1)
  %param_2.831 = s32[] parameter(2)
  %constant_5154 = s32[] constant(13)
  %add.1431 = s32[] add(s32[] %param_2.831, s32[] %constant_5154), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1059 = pred[] compare(s32[] %add.1431, s32[] %constant_4387), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5152 = s32[] constant(88)
  %add.1430 = s32[] add(s32[] %param_2.831, s32[] %constant_5152), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.571 = s32[] select(pred[] %compare.1059, s32[] %add.1430, s32[] %add.1431), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.689 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.725, s32[] %select.571), dynamic_slice_sizes={1}
  %bitcast.217 = s32[] bitcast(s32[1]{0} %dynamic-slice.689), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1058 = pred[] compare(s32[] %bitcast.217, s32[] %constant_4387), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5150 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1429 = s32[] add(s32[] %bitcast.217, s32[] %constant_5150), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.570 = s32[] select(pred[] %compare.1058, s32[] %add.1429, s32[] %bitcast.217), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.677 = pred[] compare(s32[] %constant_4387, s32[] %select.570), direction=LE
  %constant_4386 = s32[] constant(74)
  %compare.676 = pred[] compare(s32[] %constant_4386, s32[] %select.570), direction=GE
  %and.252 = pred[] and(pred[] %compare.677, pred[] %compare.676)
  %bitcast.87 = pred[1]{0} bitcast(pred[] %and.252)
  %param_3.868 = s32[75]{0} parameter(3)
  %dynamic-slice.810 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.868, s32[] %select.571), dynamic_slice_sizes={1}
  %dynamic-slice.545 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.16, s32[] %select.570), dynamic_slice_sizes={1}
  %select.238 = s32[1]{0} select(pred[1]{0} %bitcast.87, s32[1]{0} %dynamic-slice.810, s32[1]{0} %dynamic-slice.545)
  ROOT %dynamic-update-slice.225 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.16, s32[1]{0} %select.238, s32[] %select.570), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.9 (param_0.17: s32[75], param_1.723: s32[], param_2.830: s32[75]) -> s32[75] {
  %param_0.17 = s32[75]{0} parameter(0)
  %constant_4389 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.723 = s32[] parameter(1)
  %constant_5132 = s32[] constant(13)
  %add.1417 = s32[] add(s32[] %param_1.723, s32[] %constant_5132), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1050 = pred[] compare(s32[] %add.1417, s32[] %constant_4389), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5130 = s32[] constant(88)
  %add.1416 = s32[] add(s32[] %param_1.723, s32[] %constant_5130), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.562 = s32[] select(pred[] %compare.1050, s32[] %add.1416, s32[] %add.1417), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.679 = pred[] compare(s32[] %constant_4389, s32[] %select.562), direction=LE
  %constant_4388 = s32[] constant(74)
  %compare.678 = pred[] compare(s32[] %constant_4388, s32[] %select.562), direction=GE
  %and.253 = pred[] and(pred[] %compare.679, pred[] %compare.678)
  %bitcast.88 = pred[1]{0} bitcast(pred[] %and.253)
  %param_2.830 = s32[75]{0} parameter(2)
  %dynamic-slice.687 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.830, s32[] %select.562), dynamic_slice_sizes={1}
  %bitcast.215 = s32[] bitcast(s32[1]{0} %dynamic-slice.687), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1053 = pred[] compare(s32[] %bitcast.215, s32[] %constant_4389), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5139 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1422 = s32[] add(s32[] %bitcast.215, s32[] %constant_5139), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.565 = s32[] select(pred[] %compare.1053, s32[] %add.1422, s32[] %bitcast.215), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.546 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.17, s32[] %select.565), dynamic_slice_sizes={1}
  %dynamic-slice.808 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.17, s32[] %select.562), dynamic_slice_sizes={1}
  %select.239 = s32[1]{0} select(pred[1]{0} %bitcast.88, s32[1]{0} %dynamic-slice.546, s32[1]{0} %dynamic-slice.808)
  ROOT %dynamic-update-slice.226 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.17, s32[1]{0} %select.239, s32[] %select.562), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.12 (param_0.24: s32[75], param_1.721: s32[75], param_2.829: s32[], param_3.863: s32[75]) -> s32[75] {
  %param_0.24 = s32[75]{0} parameter(0)
  %constant_4394 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.721 = s32[75]{0} parameter(1)
  %param_2.829 = s32[] parameter(2)
  %constant_5111 = s32[] constant(12)
  %add.1403 = s32[] add(s32[] %param_2.829, s32[] %constant_5111), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1043 = pred[] compare(s32[] %add.1403, s32[] %constant_4394), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5109 = s32[] constant(87)
  %add.1402 = s32[] add(s32[] %param_2.829, s32[] %constant_5109), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.555 = s32[] select(pred[] %compare.1043, s32[] %add.1402, s32[] %add.1403), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.684 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.721, s32[] %select.555), dynamic_slice_sizes={1}
  %bitcast.213 = s32[] bitcast(s32[1]{0} %dynamic-slice.684), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1042 = pred[] compare(s32[] %bitcast.213, s32[] %constant_4394), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5107 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1401 = s32[] add(s32[] %bitcast.213, s32[] %constant_5107), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.554 = s32[] select(pred[] %compare.1042, s32[] %add.1401, s32[] %bitcast.213), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.684 = pred[] compare(s32[] %constant_4394, s32[] %select.554), direction=LE
  %constant_4393 = s32[] constant(74)
  %compare.683 = pred[] compare(s32[] %constant_4393, s32[] %select.554), direction=GE
  %and.254 = pred[] and(pred[] %compare.684, pred[] %compare.683)
  %bitcast.90 = pred[1]{0} bitcast(pred[] %and.254)
  %param_3.863 = s32[75]{0} parameter(3)
  %dynamic-slice.806 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.863, s32[] %select.555), dynamic_slice_sizes={1}
  %dynamic-slice.548 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.24, s32[] %select.554), dynamic_slice_sizes={1}
  %select.242 = s32[1]{0} select(pred[1]{0} %bitcast.90, s32[1]{0} %dynamic-slice.806, s32[1]{0} %dynamic-slice.548)
  ROOT %dynamic-update-slice.227 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.24, s32[1]{0} %select.242, s32[] %select.554), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.13 (param_0.25: s32[75], param_1.719: s32[], param_2.828: s32[75]) -> s32[75] {
  %param_0.25 = s32[75]{0} parameter(0)
  %constant_4396 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.719 = s32[] parameter(1)
  %constant_5089 = s32[] constant(12)
  %add.1390 = s32[] add(s32[] %param_1.719, s32[] %constant_5089), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1035 = pred[] compare(s32[] %add.1390, s32[] %constant_4396), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5087 = s32[] constant(87)
  %add.1389 = s32[] add(s32[] %param_1.719, s32[] %constant_5087), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.546 = s32[] select(pred[] %compare.1035, s32[] %add.1389, s32[] %add.1390), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.686 = pred[] compare(s32[] %constant_4396, s32[] %select.546), direction=LE
  %constant_4395 = s32[] constant(74)
  %compare.685 = pred[] compare(s32[] %constant_4395, s32[] %select.546), direction=GE
  %and.255 = pred[] and(pred[] %compare.686, pred[] %compare.685)
  %bitcast.91 = pred[1]{0} bitcast(pred[] %and.255)
  %param_2.828 = s32[75]{0} parameter(2)
  %dynamic-slice.682 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.828, s32[] %select.546), dynamic_slice_sizes={1}
  %bitcast.211 = s32[] bitcast(s32[1]{0} %dynamic-slice.682), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1038 = pred[] compare(s32[] %bitcast.211, s32[] %constant_4396), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5096 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1394 = s32[] add(s32[] %bitcast.211, s32[] %constant_5096), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.549 = s32[] select(pred[] %compare.1038, s32[] %add.1394, s32[] %bitcast.211), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.549 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.25, s32[] %select.549), dynamic_slice_sizes={1}
  %dynamic-slice.804 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.25, s32[] %select.546), dynamic_slice_sizes={1}
  %select.243 = s32[1]{0} select(pred[1]{0} %bitcast.91, s32[1]{0} %dynamic-slice.549, s32[1]{0} %dynamic-slice.804)
  ROOT %dynamic-update-slice.228 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.25, s32[1]{0} %select.243, s32[] %select.546), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.16 (param_0.32: s32[75], param_1.717: s32[75], param_2.827: s32[], param_3.858: s32[75]) -> s32[75] {
  %param_0.32 = s32[75]{0} parameter(0)
  %constant_4401 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.717 = s32[75]{0} parameter(1)
  %param_2.827 = s32[] parameter(2)
  %constant_5068 = s32[] constant(11)
  %add.1377 = s32[] add(s32[] %param_2.827, s32[] %constant_5068), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1028 = pred[] compare(s32[] %add.1377, s32[] %constant_4401), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5066 = s32[] constant(86)
  %add.1375 = s32[] add(s32[] %param_2.827, s32[] %constant_5066), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.539 = s32[] select(pred[] %compare.1028, s32[] %add.1375, s32[] %add.1377), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.679 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.717, s32[] %select.539), dynamic_slice_sizes={1}
  %bitcast.209 = s32[] bitcast(s32[1]{0} %dynamic-slice.679), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1027 = pred[] compare(s32[] %bitcast.209, s32[] %constant_4401), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5064 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1374 = s32[] add(s32[] %bitcast.209, s32[] %constant_5064), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.538 = s32[] select(pred[] %compare.1027, s32[] %add.1374, s32[] %bitcast.209), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.691 = pred[] compare(s32[] %constant_4401, s32[] %select.538), direction=LE
  %constant_4400 = s32[] constant(74)
  %compare.689 = pred[] compare(s32[] %constant_4400, s32[] %select.538), direction=GE
  %and.256 = pred[] and(pred[] %compare.691, pred[] %compare.689)
  %bitcast.93 = pred[1]{0} bitcast(pred[] %and.256)
  %param_3.858 = s32[75]{0} parameter(3)
  %dynamic-slice.802 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.858, s32[] %select.539), dynamic_slice_sizes={1}
  %dynamic-slice.551 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.32, s32[] %select.538), dynamic_slice_sizes={1}
  %select.247 = s32[1]{0} select(pred[1]{0} %bitcast.93, s32[1]{0} %dynamic-slice.802, s32[1]{0} %dynamic-slice.551)
  ROOT %dynamic-update-slice.229 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.32, s32[1]{0} %select.247, s32[] %select.538), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.17 (param_0.33: s32[75], param_1.715: s32[], param_2.826: s32[75]) -> s32[75] {
  %param_0.33 = s32[75]{0} parameter(0)
  %constant_4403 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.715 = s32[] parameter(1)
  %constant_5046 = s32[] constant(11)
  %add.1363 = s32[] add(s32[] %param_1.715, s32[] %constant_5046), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1020 = pred[] compare(s32[] %add.1363, s32[] %constant_4403), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5044 = s32[] constant(86)
  %add.1362 = s32[] add(s32[] %param_1.715, s32[] %constant_5044), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.530 = s32[] select(pred[] %compare.1020, s32[] %add.1362, s32[] %add.1363), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.693 = pred[] compare(s32[] %constant_4403, s32[] %select.530), direction=LE
  %constant_4402 = s32[] constant(74)
  %compare.692 = pred[] compare(s32[] %constant_4402, s32[] %select.530), direction=GE
  %and.257 = pred[] and(pred[] %compare.693, pred[] %compare.692)
  %bitcast.94 = pred[1]{0} bitcast(pred[] %and.257)
  %param_2.826 = s32[75]{0} parameter(2)
  %dynamic-slice.677 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.826, s32[] %select.530), dynamic_slice_sizes={1}
  %bitcast.207 = s32[] bitcast(s32[1]{0} %dynamic-slice.677), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1023 = pred[] compare(s32[] %bitcast.207, s32[] %constant_4403), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5053 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1367 = s32[] add(s32[] %bitcast.207, s32[] %constant_5053), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.534 = s32[] select(pred[] %compare.1023, s32[] %add.1367, s32[] %bitcast.207), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.552 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.33, s32[] %select.534), dynamic_slice_sizes={1}
  %dynamic-slice.800 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.33, s32[] %select.530), dynamic_slice_sizes={1}
  %select.248 = s32[1]{0} select(pred[1]{0} %bitcast.94, s32[1]{0} %dynamic-slice.552, s32[1]{0} %dynamic-slice.800)
  ROOT %dynamic-update-slice.230 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.33, s32[1]{0} %select.248, s32[] %select.530), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.20 (param_0.40: s32[75], param_1.713: s32[75], param_2.825: s32[], param_3.853: s32[75]) -> s32[75] {
  %param_0.40 = s32[75]{0} parameter(0)
  %constant_4408 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.713 = s32[75]{0} parameter(1)
  %param_2.825 = s32[] parameter(2)
  %constant_5025 = s32[] constant(10)
  %add.1349 = s32[] add(s32[] %param_2.825, s32[] %constant_5025), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1013 = pred[] compare(s32[] %add.1349, s32[] %constant_4408), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5023 = s32[] constant(85)
  %add.1348 = s32[] add(s32[] %param_2.825, s32[] %constant_5023), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.523 = s32[] select(pred[] %compare.1013, s32[] %add.1348, s32[] %add.1349), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.674 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.713, s32[] %select.523), dynamic_slice_sizes={1}
  %bitcast.205 = s32[] bitcast(s32[1]{0} %dynamic-slice.674), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1012 = pred[] compare(s32[] %bitcast.205, s32[] %constant_4408), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5021 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1347 = s32[] add(s32[] %bitcast.205, s32[] %constant_5021), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.522 = s32[] select(pred[] %compare.1012, s32[] %add.1347, s32[] %bitcast.205), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.697 = pred[] compare(s32[] %constant_4408, s32[] %select.522), direction=LE
  %constant_4407 = s32[] constant(74)
  %compare.696 = pred[] compare(s32[] %constant_4407, s32[] %select.522), direction=GE
  %and.258 = pred[] and(pred[] %compare.697, pred[] %compare.696)
  %bitcast.96 = pred[1]{0} bitcast(pred[] %and.258)
  %param_3.853 = s32[75]{0} parameter(3)
  %dynamic-slice.798 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.853, s32[] %select.523), dynamic_slice_sizes={1}
  %dynamic-slice.554 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.40, s32[] %select.522), dynamic_slice_sizes={1}
  %select.251 = s32[1]{0} select(pred[1]{0} %bitcast.96, s32[1]{0} %dynamic-slice.798, s32[1]{0} %dynamic-slice.554)
  ROOT %dynamic-update-slice.231 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.40, s32[1]{0} %select.251, s32[] %select.522), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.21 (param_0.41: s32[75], param_1.711: s32[], param_2.824: s32[75]) -> s32[75] {
  %param_0.41 = s32[75]{0} parameter(0)
  %constant_4410 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.711 = s32[] parameter(1)
  %constant_5003 = s32[] constant(10)
  %add.1336 = s32[] add(s32[] %param_1.711, s32[] %constant_5003), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1004 = pred[] compare(s32[] %add.1336, s32[] %constant_4410), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5001 = s32[] constant(85)
  %add.1335 = s32[] add(s32[] %param_1.711, s32[] %constant_5001), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.514 = s32[] select(pred[] %compare.1004, s32[] %add.1335, s32[] %add.1336), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.700 = pred[] compare(s32[] %constant_4410, s32[] %select.514), direction=LE
  %constant_4409 = s32[] constant(74)
  %compare.699 = pred[] compare(s32[] %constant_4409, s32[] %select.514), direction=GE
  %and.259 = pred[] and(pred[] %compare.700, pred[] %compare.699)
  %bitcast.97 = pred[1]{0} bitcast(pred[] %and.259)
  %param_2.824 = s32[75]{0} parameter(2)
  %dynamic-slice.672 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.824, s32[] %select.514), dynamic_slice_sizes={1}
  %bitcast.203 = s32[] bitcast(s32[1]{0} %dynamic-slice.672), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1007 = pred[] compare(s32[] %bitcast.203, s32[] %constant_4410), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5010 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1340 = s32[] add(s32[] %bitcast.203, s32[] %constant_5010), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.518 = s32[] select(pred[] %compare.1007, s32[] %add.1340, s32[] %bitcast.203), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.555 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.41, s32[] %select.518), dynamic_slice_sizes={1}
  %dynamic-slice.796 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.41, s32[] %select.514), dynamic_slice_sizes={1}
  %select.252 = s32[1]{0} select(pred[1]{0} %bitcast.97, s32[1]{0} %dynamic-slice.555, s32[1]{0} %dynamic-slice.796)
  ROOT %dynamic-update-slice.232 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.41, s32[1]{0} %select.252, s32[] %select.514), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.24 (param_0.48: s32[75], param_1.709: s32[75], param_2.823: s32[], param_3.848: s32[75]) -> s32[75] {
  %param_0.48 = s32[75]{0} parameter(0)
  %constant_4415 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.709 = s32[75]{0} parameter(1)
  %param_2.823 = s32[] parameter(2)
  %constant_4982 = s32[] constant(9)
  %add.1322 = s32[] add(s32[] %param_2.823, s32[] %constant_4982), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.997 = pred[] compare(s32[] %add.1322, s32[] %constant_4415), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4980 = s32[] constant(84)
  %add.1321 = s32[] add(s32[] %param_2.823, s32[] %constant_4980), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.507 = s32[] select(pred[] %compare.997, s32[] %add.1321, s32[] %add.1322), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.669 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.709, s32[] %select.507), dynamic_slice_sizes={1}
  %bitcast.201 = s32[] bitcast(s32[1]{0} %dynamic-slice.669), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.996 = pred[] compare(s32[] %bitcast.201, s32[] %constant_4415), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4978 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1320 = s32[] add(s32[] %bitcast.201, s32[] %constant_4978), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.506 = s32[] select(pred[] %compare.996, s32[] %add.1320, s32[] %bitcast.201), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.704 = pred[] compare(s32[] %constant_4415, s32[] %select.506), direction=LE
  %constant_4414 = s32[] constant(74)
  %compare.703 = pred[] compare(s32[] %constant_4414, s32[] %select.506), direction=GE
  %and.260 = pred[] and(pred[] %compare.704, pred[] %compare.703)
  %bitcast.99 = pred[1]{0} bitcast(pred[] %and.260)
  %param_3.848 = s32[75]{0} parameter(3)
  %dynamic-slice.794 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.848, s32[] %select.507), dynamic_slice_sizes={1}
  %dynamic-slice.557 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.48, s32[] %select.506), dynamic_slice_sizes={1}
  %select.256 = s32[1]{0} select(pred[1]{0} %bitcast.99, s32[1]{0} %dynamic-slice.794, s32[1]{0} %dynamic-slice.557)
  ROOT %dynamic-update-slice.233 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.48, s32[1]{0} %select.256, s32[] %select.506), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.25 (param_0.49: s32[75], param_1.707: s32[], param_2.822: s32[75]) -> s32[75] {
  %param_0.49 = s32[75]{0} parameter(0)
  %constant_4417 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.707 = s32[] parameter(1)
  %constant_4960 = s32[] constant(9)
  %add.1309 = s32[] add(s32[] %param_1.707, s32[] %constant_4960), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.988 = pred[] compare(s32[] %add.1309, s32[] %constant_4417), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4958 = s32[] constant(84)
  %add.1307 = s32[] add(s32[] %param_1.707, s32[] %constant_4958), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.499 = s32[] select(pred[] %compare.988, s32[] %add.1307, s32[] %add.1309), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.707 = pred[] compare(s32[] %constant_4417, s32[] %select.499), direction=LE
  %constant_4416 = s32[] constant(74)
  %compare.705 = pred[] compare(s32[] %constant_4416, s32[] %select.499), direction=GE
  %and.261 = pred[] and(pred[] %compare.707, pred[] %compare.705)
  %bitcast.100 = pred[1]{0} bitcast(pred[] %and.261)
  %param_2.822 = s32[75]{0} parameter(2)
  %dynamic-slice.667 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.822, s32[] %select.499), dynamic_slice_sizes={1}
  %bitcast.199 = s32[] bitcast(s32[1]{0} %dynamic-slice.667), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.991 = pred[] compare(s32[] %bitcast.199, s32[] %constant_4417), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4967 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1313 = s32[] add(s32[] %bitcast.199, s32[] %constant_4967), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.502 = s32[] select(pred[] %compare.991, s32[] %add.1313, s32[] %bitcast.199), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.558 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.49, s32[] %select.502), dynamic_slice_sizes={1}
  %dynamic-slice.792 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.49, s32[] %select.499), dynamic_slice_sizes={1}
  %select.257 = s32[1]{0} select(pred[1]{0} %bitcast.100, s32[1]{0} %dynamic-slice.558, s32[1]{0} %dynamic-slice.792)
  ROOT %dynamic-update-slice.234 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.49, s32[1]{0} %select.257, s32[] %select.499), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.28 (param_0.56: s32[75], param_1.705: s32[75], param_2.821: s32[], param_3.843: s32[75]) -> s32[75] {
  %param_0.56 = s32[75]{0} parameter(0)
  %constant_4422 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.705 = s32[75]{0} parameter(1)
  %param_2.821 = s32[] parameter(2)
  %constant_4939 = s32[] constant(8)
  %add.1295 = s32[] add(s32[] %param_2.821, s32[] %constant_4939), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.982 = pred[] compare(s32[] %add.1295, s32[] %constant_4422), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4937 = s32[] constant(83)
  %add.1294 = s32[] add(s32[] %param_2.821, s32[] %constant_4937), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.492 = s32[] select(pred[] %compare.982, s32[] %add.1294, s32[] %add.1295), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.664 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.705, s32[] %select.492), dynamic_slice_sizes={1}
  %bitcast.197 = s32[] bitcast(s32[1]{0} %dynamic-slice.664), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.981 = pred[] compare(s32[] %bitcast.197, s32[] %constant_4422), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4935 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1292 = s32[] add(s32[] %bitcast.197, s32[] %constant_4935), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.491 = s32[] select(pred[] %compare.981, s32[] %add.1292, s32[] %bitcast.197), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.711 = pred[] compare(s32[] %constant_4422, s32[] %select.491), direction=LE
  %constant_4421 = s32[] constant(74)
  %compare.710 = pred[] compare(s32[] %constant_4421, s32[] %select.491), direction=GE
  %and.262 = pred[] and(pred[] %compare.711, pred[] %compare.710)
  %bitcast.102 = pred[1]{0} bitcast(pred[] %and.262)
  %param_3.843 = s32[75]{0} parameter(3)
  %dynamic-slice.790 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.843, s32[] %select.492), dynamic_slice_sizes={1}
  %dynamic-slice.560 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.56, s32[] %select.491), dynamic_slice_sizes={1}
  %select.260 = s32[1]{0} select(pred[1]{0} %bitcast.102, s32[1]{0} %dynamic-slice.790, s32[1]{0} %dynamic-slice.560)
  ROOT %dynamic-update-slice.235 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.56, s32[1]{0} %select.260, s32[] %select.491), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.29 (param_0.57: s32[75], param_1.703: s32[], param_2.820: s32[75]) -> s32[75] {
  %param_0.57 = s32[75]{0} parameter(0)
  %constant_4424 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.703 = s32[] parameter(1)
  %constant_4917 = s32[] constant(8)
  %add.1282 = s32[] add(s32[] %param_1.703, s32[] %constant_4917), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.973 = pred[] compare(s32[] %add.1282, s32[] %constant_4424), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4915 = s32[] constant(83)
  %add.1281 = s32[] add(s32[] %param_1.703, s32[] %constant_4915), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.483 = s32[] select(pred[] %compare.973, s32[] %add.1281, s32[] %add.1282), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.713 = pred[] compare(s32[] %constant_4424, s32[] %select.483), direction=LE
  %constant_4423 = s32[] constant(74)
  %compare.712 = pred[] compare(s32[] %constant_4423, s32[] %select.483), direction=GE
  %and.263 = pred[] and(pred[] %compare.713, pred[] %compare.712)
  %bitcast.103 = pred[1]{0} bitcast(pred[] %and.263)
  %param_2.820 = s32[75]{0} parameter(2)
  %dynamic-slice.662 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.820, s32[] %select.483), dynamic_slice_sizes={1}
  %bitcast.195 = s32[] bitcast(s32[1]{0} %dynamic-slice.662), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.976 = pred[] compare(s32[] %bitcast.195, s32[] %constant_4424), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4924 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1286 = s32[] add(s32[] %bitcast.195, s32[] %constant_4924), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.486 = s32[] select(pred[] %compare.976, s32[] %add.1286, s32[] %bitcast.195), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.561 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.57, s32[] %select.486), dynamic_slice_sizes={1}
  %dynamic-slice.788 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.57, s32[] %select.483), dynamic_slice_sizes={1}
  %select.262 = s32[1]{0} select(pred[1]{0} %bitcast.103, s32[1]{0} %dynamic-slice.561, s32[1]{0} %dynamic-slice.788)
  ROOT %dynamic-update-slice.236 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.57, s32[1]{0} %select.262, s32[] %select.483), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.32 (param_0.64: s32[75], param_1.701: s32[75], param_2.819: s32[], param_3.838: s32[75]) -> s32[75] {
  %param_0.64 = s32[75]{0} parameter(0)
  %constant_4429 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.701 = s32[75]{0} parameter(1)
  %param_2.819 = s32[] parameter(2)
  %constant_4896 = s32[] constant(7)
  %add.1268 = s32[] add(s32[] %param_2.819, s32[] %constant_4896), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.967 = pred[] compare(s32[] %add.1268, s32[] %constant_4429), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4894 = s32[] constant(82)
  %add.1267 = s32[] add(s32[] %param_2.819, s32[] %constant_4894), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.476 = s32[] select(pred[] %compare.967, s32[] %add.1267, s32[] %add.1268), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.659 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.701, s32[] %select.476), dynamic_slice_sizes={1}
  %bitcast.193 = s32[] bitcast(s32[1]{0} %dynamic-slice.659), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.965 = pred[] compare(s32[] %bitcast.193, s32[] %constant_4429), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4892 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1266 = s32[] add(s32[] %bitcast.193, s32[] %constant_4892), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.475 = s32[] select(pred[] %compare.965, s32[] %add.1266, s32[] %bitcast.193), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.718 = pred[] compare(s32[] %constant_4429, s32[] %select.475), direction=LE
  %constant_4428 = s32[] constant(74)
  %compare.717 = pred[] compare(s32[] %constant_4428, s32[] %select.475), direction=GE
  %and.264 = pred[] and(pred[] %compare.718, pred[] %compare.717)
  %bitcast.105 = pred[1]{0} bitcast(pred[] %and.264)
  %param_3.838 = s32[75]{0} parameter(3)
  %dynamic-slice.786 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.838, s32[] %select.476), dynamic_slice_sizes={1}
  %dynamic-slice.563 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.64, s32[] %select.475), dynamic_slice_sizes={1}
  %select.265 = s32[1]{0} select(pred[1]{0} %bitcast.105, s32[1]{0} %dynamic-slice.786, s32[1]{0} %dynamic-slice.563)
  ROOT %dynamic-update-slice.237 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.64, s32[1]{0} %select.265, s32[] %select.475), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.33 (param_0.65: s32[75], param_1.699: s32[], param_2.818: s32[75]) -> s32[75] {
  %param_0.65 = s32[75]{0} parameter(0)
  %constant_4431 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.699 = s32[] parameter(1)
  %constant_4874 = s32[] constant(7)
  %add.1254 = s32[] add(s32[] %param_1.699, s32[] %constant_4874), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.957 = pred[] compare(s32[] %add.1254, s32[] %constant_4431), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4872 = s32[] constant(82)
  %add.1253 = s32[] add(s32[] %param_1.699, s32[] %constant_4872), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.467 = s32[] select(pred[] %compare.957, s32[] %add.1253, s32[] %add.1254), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.720 = pred[] compare(s32[] %constant_4431, s32[] %select.467), direction=LE
  %constant_4430 = s32[] constant(74)
  %compare.719 = pred[] compare(s32[] %constant_4430, s32[] %select.467), direction=GE
  %and.265 = pred[] and(pred[] %compare.720, pred[] %compare.719)
  %bitcast.106 = pred[1]{0} bitcast(pred[] %and.265)
  %param_2.818 = s32[75]{0} parameter(2)
  %dynamic-slice.657 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.818, s32[] %select.467), dynamic_slice_sizes={1}
  %bitcast.191 = s32[] bitcast(s32[1]{0} %dynamic-slice.657), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.961 = pred[] compare(s32[] %bitcast.191, s32[] %constant_4431), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4881 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1259 = s32[] add(s32[] %bitcast.191, s32[] %constant_4881), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.470 = s32[] select(pred[] %compare.961, s32[] %add.1259, s32[] %bitcast.191), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.564 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.65, s32[] %select.470), dynamic_slice_sizes={1}
  %dynamic-slice.784 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.65, s32[] %select.467), dynamic_slice_sizes={1}
  %select.266 = s32[1]{0} select(pred[1]{0} %bitcast.106, s32[1]{0} %dynamic-slice.564, s32[1]{0} %dynamic-slice.784)
  ROOT %dynamic-update-slice.238 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.65, s32[1]{0} %select.266, s32[] %select.467), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.36 (param_0.72: s32[75], param_1.697: s32[75], param_2.817: s32[], param_3.833: s32[75]) -> s32[75] {
  %param_0.72 = s32[75]{0} parameter(0)
  %constant_4436 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.697 = s32[75]{0} parameter(1)
  %param_2.817 = s32[] parameter(2)
  %constant_4853 = s32[] constant(6)
  %add.1240 = s32[] add(s32[] %param_2.817, s32[] %constant_4853), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.951 = pred[] compare(s32[] %add.1240, s32[] %constant_4436), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4851 = s32[] constant(81)
  %add.1239 = s32[] add(s32[] %param_2.817, s32[] %constant_4851), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.460 = s32[] select(pred[] %compare.951, s32[] %add.1239, s32[] %add.1240), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.654 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.697, s32[] %select.460), dynamic_slice_sizes={1}
  %bitcast.189 = s32[] bitcast(s32[1]{0} %dynamic-slice.654), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.949 = pred[] compare(s32[] %bitcast.189, s32[] %constant_4436), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4849 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1238 = s32[] add(s32[] %bitcast.189, s32[] %constant_4849), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.459 = s32[] select(pred[] %compare.949, s32[] %add.1238, s32[] %bitcast.189), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.724 = pred[] compare(s32[] %constant_4436, s32[] %select.459), direction=LE
  %constant_4435 = s32[] constant(74)
  %compare.723 = pred[] compare(s32[] %constant_4435, s32[] %select.459), direction=GE
  %and.266 = pred[] and(pred[] %compare.724, pred[] %compare.723)
  %bitcast.108 = pred[1]{0} bitcast(pred[] %and.266)
  %param_3.833 = s32[75]{0} parameter(3)
  %dynamic-slice.782 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.833, s32[] %select.460), dynamic_slice_sizes={1}
  %dynamic-slice.566 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.72, s32[] %select.459), dynamic_slice_sizes={1}
  %select.270 = s32[1]{0} select(pred[1]{0} %bitcast.108, s32[1]{0} %dynamic-slice.782, s32[1]{0} %dynamic-slice.566)
  ROOT %dynamic-update-slice.239 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.72, s32[1]{0} %select.270, s32[] %select.459), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.37 (param_0.73: s32[75], param_1.695: s32[], param_2.816: s32[75]) -> s32[75] {
  %param_0.73 = s32[75]{0} parameter(0)
  %constant_4438 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.695 = s32[] parameter(1)
  %constant_4831 = s32[] constant(6)
  %add.1227 = s32[] add(s32[] %param_1.695, s32[] %constant_4831), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.941 = pred[] compare(s32[] %add.1227, s32[] %constant_4438), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4829 = s32[] constant(81)
  %add.1226 = s32[] add(s32[] %param_1.695, s32[] %constant_4829), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.451 = s32[] select(pred[] %compare.941, s32[] %add.1226, s32[] %add.1227), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.727 = pred[] compare(s32[] %constant_4438, s32[] %select.451), direction=LE
  %constant_4437 = s32[] constant(74)
  %compare.725 = pred[] compare(s32[] %constant_4437, s32[] %select.451), direction=GE
  %and.267 = pred[] and(pred[] %compare.727, pred[] %compare.725)
  %bitcast.109 = pred[1]{0} bitcast(pred[] %and.267)
  %param_2.816 = s32[75]{0} parameter(2)
  %dynamic-slice.652 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.816, s32[] %select.451), dynamic_slice_sizes={1}
  %bitcast.187 = s32[] bitcast(s32[1]{0} %dynamic-slice.652), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.945 = pred[] compare(s32[] %bitcast.187, s32[] %constant_4438), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4838 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1232 = s32[] add(s32[] %bitcast.187, s32[] %constant_4838), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.455 = s32[] select(pred[] %compare.945, s32[] %add.1232, s32[] %bitcast.187), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.567 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.73, s32[] %select.455), dynamic_slice_sizes={1}
  %dynamic-slice.780 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.73, s32[] %select.451), dynamic_slice_sizes={1}
  %select.271 = s32[1]{0} select(pred[1]{0} %bitcast.109, s32[1]{0} %dynamic-slice.567, s32[1]{0} %dynamic-slice.780)
  ROOT %dynamic-update-slice.240 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.73, s32[1]{0} %select.271, s32[] %select.451), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.40 (param_0.80: s32[75], param_1.693: s32[75], param_2.815: s32[], param_3.828: s32[75]) -> s32[75] {
  %param_0.80 = s32[75]{0} parameter(0)
  %constant_4443 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.693 = s32[75]{0} parameter(1)
  %param_2.815 = s32[] parameter(2)
  %constant_4810 = s32[] constant(5)
  %add.1214 = s32[] add(s32[] %param_2.815, s32[] %constant_4810), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.935 = pred[] compare(s32[] %add.1214, s32[] %constant_4443), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4808 = s32[] constant(80)
  %add.1213 = s32[] add(s32[] %param_2.815, s32[] %constant_4808), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.444 = s32[] select(pred[] %compare.935, s32[] %add.1213, s32[] %add.1214), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.649 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.693, s32[] %select.444), dynamic_slice_sizes={1}
  %bitcast.185 = s32[] bitcast(s32[1]{0} %dynamic-slice.649), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.934 = pred[] compare(s32[] %bitcast.185, s32[] %constant_4443), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4806 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1211 = s32[] add(s32[] %bitcast.185, s32[] %constant_4806), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.443 = s32[] select(pred[] %compare.934, s32[] %add.1211, s32[] %bitcast.185), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.731 = pred[] compare(s32[] %constant_4443, s32[] %select.443), direction=LE
  %constant_4442 = s32[] constant(74)
  %compare.730 = pred[] compare(s32[] %constant_4442, s32[] %select.443), direction=GE
  %and.268 = pred[] and(pred[] %compare.731, pred[] %compare.730)
  %bitcast.111 = pred[1]{0} bitcast(pred[] %and.268)
  %param_3.828 = s32[75]{0} parameter(3)
  %dynamic-slice.778 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.828, s32[] %select.444), dynamic_slice_sizes={1}
  %dynamic-slice.569 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.80, s32[] %select.443), dynamic_slice_sizes={1}
  %select.274 = s32[1]{0} select(pred[1]{0} %bitcast.111, s32[1]{0} %dynamic-slice.778, s32[1]{0} %dynamic-slice.569)
  ROOT %dynamic-update-slice.241 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.80, s32[1]{0} %select.274, s32[] %select.443), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.41 (param_0.81: s32[75], param_1.691: s32[], param_2.814: s32[75]) -> s32[75] {
  %param_0.81 = s32[75]{0} parameter(0)
  %constant_4445 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.691 = s32[] parameter(1)
  %constant_4788 = s32[] constant(5)
  %add.1199 = s32[] add(s32[] %param_1.691, s32[] %constant_4788), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.926 = pred[] compare(s32[] %add.1199, s32[] %constant_4445), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4786 = s32[] constant(80)
  %add.1198 = s32[] add(s32[] %param_1.691, s32[] %constant_4786), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.435 = s32[] select(pred[] %compare.926, s32[] %add.1198, s32[] %add.1199), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.733 = pred[] compare(s32[] %constant_4445, s32[] %select.435), direction=LE
  %constant_4444 = s32[] constant(74)
  %compare.732 = pred[] compare(s32[] %constant_4444, s32[] %select.435), direction=GE
  %and.269 = pred[] and(pred[] %compare.733, pred[] %compare.732)
  %bitcast.112 = pred[1]{0} bitcast(pred[] %and.269)
  %param_2.814 = s32[75]{0} parameter(2)
  %dynamic-slice.647 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.814, s32[] %select.435), dynamic_slice_sizes={1}
  %bitcast.183 = s32[] bitcast(s32[1]{0} %dynamic-slice.647), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.930 = pred[] compare(s32[] %bitcast.183, s32[] %constant_4445), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4795 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1205 = s32[] add(s32[] %bitcast.183, s32[] %constant_4795), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.439 = s32[] select(pred[] %compare.930, s32[] %add.1205, s32[] %bitcast.183), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.570 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.81, s32[] %select.439), dynamic_slice_sizes={1}
  %dynamic-slice.776 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.81, s32[] %select.435), dynamic_slice_sizes={1}
  %select.275 = s32[1]{0} select(pred[1]{0} %bitcast.112, s32[1]{0} %dynamic-slice.570, s32[1]{0} %dynamic-slice.776)
  ROOT %dynamic-update-slice.242 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.81, s32[1]{0} %select.275, s32[] %select.435), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.44 (param_0.88: s32[75], param_1.689: s32[75], param_2.813: s32[], param_3.823: s32[75]) -> s32[75] {
  %param_0.88 = s32[75]{0} parameter(0)
  %constant_4450 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.689 = s32[75]{0} parameter(1)
  %param_2.813 = s32[] parameter(2)
  %constant_4767 = s32[] constant(4)
  %add.1186 = s32[] add(s32[] %param_2.813, s32[] %constant_4767), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.920 = pred[] compare(s32[] %add.1186, s32[] %constant_4450), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4765 = s32[] constant(79)
  %add.1185 = s32[] add(s32[] %param_2.813, s32[] %constant_4765), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.428 = s32[] select(pred[] %compare.920, s32[] %add.1185, s32[] %add.1186), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.644 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.689, s32[] %select.428), dynamic_slice_sizes={1}
  %bitcast.181 = s32[] bitcast(s32[1]{0} %dynamic-slice.644), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.919 = pred[] compare(s32[] %bitcast.181, s32[] %constant_4450), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4763 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1184 = s32[] add(s32[] %bitcast.181, s32[] %constant_4763), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.427 = s32[] select(pred[] %compare.919, s32[] %add.1184, s32[] %bitcast.181), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.738 = pred[] compare(s32[] %constant_4450, s32[] %select.427), direction=LE
  %constant_4449 = s32[] constant(74)
  %compare.737 = pred[] compare(s32[] %constant_4449, s32[] %select.427), direction=GE
  %and.270 = pred[] and(pred[] %compare.738, pred[] %compare.737)
  %bitcast.114 = pred[1]{0} bitcast(pred[] %and.270)
  %param_3.823 = s32[75]{0} parameter(3)
  %dynamic-slice.774 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.823, s32[] %select.428), dynamic_slice_sizes={1}
  %dynamic-slice.572 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.88, s32[] %select.427), dynamic_slice_sizes={1}
  %select.279 = s32[1]{0} select(pred[1]{0} %bitcast.114, s32[1]{0} %dynamic-slice.774, s32[1]{0} %dynamic-slice.572)
  ROOT %dynamic-update-slice.243 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.88, s32[1]{0} %select.279, s32[] %select.427), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.45 (param_0.89: s32[75], param_1.687: s32[], param_2.812: s32[75]) -> s32[75] {
  %param_0.89 = s32[75]{0} parameter(0)
  %constant_4452 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.687 = s32[] parameter(1)
  %constant_4745 = s32[] constant(4)
  %add.1173 = s32[] add(s32[] %param_1.687, s32[] %constant_4745), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.911 = pred[] compare(s32[] %add.1173, s32[] %constant_4452), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4743 = s32[] constant(79)
  %add.1172 = s32[] add(s32[] %param_1.687, s32[] %constant_4743), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.419 = s32[] select(pred[] %compare.911, s32[] %add.1172, s32[] %add.1173), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.740 = pred[] compare(s32[] %constant_4452, s32[] %select.419), direction=LE
  %constant_4451 = s32[] constant(74)
  %compare.739 = pred[] compare(s32[] %constant_4451, s32[] %select.419), direction=GE
  %and.271 = pred[] and(pred[] %compare.740, pred[] %compare.739)
  %bitcast.115 = pred[1]{0} bitcast(pred[] %and.271)
  %param_2.812 = s32[75]{0} parameter(2)
  %dynamic-slice.642 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.812, s32[] %select.419), dynamic_slice_sizes={1}
  %bitcast.179 = s32[] bitcast(s32[1]{0} %dynamic-slice.642), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.915 = pred[] compare(s32[] %bitcast.179, s32[] %constant_4452), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4752 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1178 = s32[] add(s32[] %bitcast.179, s32[] %constant_4752), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.423 = s32[] select(pred[] %compare.915, s32[] %add.1178, s32[] %bitcast.179), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.573 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.89, s32[] %select.423), dynamic_slice_sizes={1}
  %dynamic-slice.772 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.89, s32[] %select.419), dynamic_slice_sizes={1}
  %select.280 = s32[1]{0} select(pred[1]{0} %bitcast.115, s32[1]{0} %dynamic-slice.573, s32[1]{0} %dynamic-slice.772)
  ROOT %dynamic-update-slice.244 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.89, s32[1]{0} %select.280, s32[] %select.419), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.48 (param_0.96: s32[75], param_1.685: s32[75], param_2.811: s32[], param_3.818: s32[75]) -> s32[75] {
  %param_0.96 = s32[75]{0} parameter(0)
  %constant_4457 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.685 = s32[75]{0} parameter(1)
  %param_2.811 = s32[] parameter(2)
  %constant_4724 = s32[] constant(3)
  %add.1159 = s32[] add(s32[] %param_2.811, s32[] %constant_4724), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.904 = pred[] compare(s32[] %add.1159, s32[] %constant_4457), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4722 = s32[] constant(78)
  %add.1158 = s32[] add(s32[] %param_2.811, s32[] %constant_4722), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.413 = s32[] select(pred[] %compare.904, s32[] %add.1158, s32[] %add.1159), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.639 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.685, s32[] %select.413), dynamic_slice_sizes={1}
  %bitcast.177 = s32[] bitcast(s32[1]{0} %dynamic-slice.639), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.903 = pred[] compare(s32[] %bitcast.177, s32[] %constant_4457), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4720 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1157 = s32[] add(s32[] %bitcast.177, s32[] %constant_4720), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.412 = s32[] select(pred[] %compare.903, s32[] %add.1157, s32[] %bitcast.177), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.745 = pred[] compare(s32[] %constant_4457, s32[] %select.412), direction=LE
  %constant_4456 = s32[] constant(74)
  %compare.744 = pred[] compare(s32[] %constant_4456, s32[] %select.412), direction=GE
  %and.272 = pred[] and(pred[] %compare.745, pred[] %compare.744)
  %bitcast.117 = pred[1]{0} bitcast(pred[] %and.272)
  %param_3.818 = s32[75]{0} parameter(3)
  %dynamic-slice.770 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.818, s32[] %select.413), dynamic_slice_sizes={1}
  %dynamic-slice.575 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.96, s32[] %select.412), dynamic_slice_sizes={1}
  %select.283 = s32[1]{0} select(pred[1]{0} %bitcast.117, s32[1]{0} %dynamic-slice.770, s32[1]{0} %dynamic-slice.575)
  ROOT %dynamic-update-slice.245 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.96, s32[1]{0} %select.283, s32[] %select.412), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.49 (param_0.97: s32[75], param_1.683: s32[], param_2.810: s32[75]) -> s32[75] {
  %param_0.97 = s32[75]{0} parameter(0)
  %constant_4459 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.683 = s32[] parameter(1)
  %constant_4702 = s32[] constant(3)
  %add.1145 = s32[] add(s32[] %param_1.683, s32[] %constant_4702), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.895 = pred[] compare(s32[] %add.1145, s32[] %constant_4459), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4700 = s32[] constant(78)
  %add.1144 = s32[] add(s32[] %param_1.683, s32[] %constant_4700), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.404 = s32[] select(pred[] %compare.895, s32[] %add.1144, s32[] %add.1145), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.747 = pred[] compare(s32[] %constant_4459, s32[] %select.404), direction=LE
  %constant_4458 = s32[] constant(74)
  %compare.746 = pred[] compare(s32[] %constant_4458, s32[] %select.404), direction=GE
  %and.273 = pred[] and(pred[] %compare.747, pred[] %compare.746)
  %bitcast.118 = pred[1]{0} bitcast(pred[] %and.273)
  %param_2.810 = s32[75]{0} parameter(2)
  %dynamic-slice.637 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.810, s32[] %select.404), dynamic_slice_sizes={1}
  %bitcast.175 = s32[] bitcast(s32[1]{0} %dynamic-slice.637), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.899 = pred[] compare(s32[] %bitcast.175, s32[] %constant_4459), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4709 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1150 = s32[] add(s32[] %bitcast.175, s32[] %constant_4709), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.407 = s32[] select(pred[] %compare.899, s32[] %add.1150, s32[] %bitcast.175), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.576 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.97, s32[] %select.407), dynamic_slice_sizes={1}
  %dynamic-slice.768 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.97, s32[] %select.404), dynamic_slice_sizes={1}
  %select.284 = s32[1]{0} select(pred[1]{0} %bitcast.118, s32[1]{0} %dynamic-slice.576, s32[1]{0} %dynamic-slice.768)
  ROOT %dynamic-update-slice.246 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.97, s32[1]{0} %select.284, s32[] %select.404), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.52 (param_0.104: s32[75], param_1.681: s32[75], param_2.809: s32[], param_3.813: s32[75]) -> s32[75] {
  %param_0.104 = s32[75]{0} parameter(0)
  %constant_4464 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.681 = s32[75]{0} parameter(1)
  %param_2.809 = s32[] parameter(2)
  %constant_4681 = s32[] constant(2)
  %add.1132 = s32[] add(s32[] %param_2.809, s32[] %constant_4681), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.888 = pred[] compare(s32[] %add.1132, s32[] %constant_4464), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4679 = s32[] constant(77)
  %add.1131 = s32[] add(s32[] %param_2.809, s32[] %constant_4679), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.397 = s32[] select(pred[] %compare.888, s32[] %add.1131, s32[] %add.1132), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.634 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.681, s32[] %select.397), dynamic_slice_sizes={1}
  %bitcast.173 = s32[] bitcast(s32[1]{0} %dynamic-slice.634), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.887 = pred[] compare(s32[] %bitcast.173, s32[] %constant_4464), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4677 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1130 = s32[] add(s32[] %bitcast.173, s32[] %constant_4677), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.396 = s32[] select(pred[] %compare.887, s32[] %add.1130, s32[] %bitcast.173), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.752 = pred[] compare(s32[] %constant_4464, s32[] %select.396), direction=LE
  %constant_4463 = s32[] constant(74)
  %compare.751 = pred[] compare(s32[] %constant_4463, s32[] %select.396), direction=GE
  %and.274 = pred[] and(pred[] %compare.752, pred[] %compare.751)
  %bitcast.120 = pred[1]{0} bitcast(pred[] %and.274)
  %param_3.813 = s32[75]{0} parameter(3)
  %dynamic-slice.766 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.813, s32[] %select.397), dynamic_slice_sizes={1}
  %dynamic-slice.578 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.104, s32[] %select.396), dynamic_slice_sizes={1}
  %select.287 = s32[1]{0} select(pred[1]{0} %bitcast.120, s32[1]{0} %dynamic-slice.766, s32[1]{0} %dynamic-slice.578)
  ROOT %dynamic-update-slice.247 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.104, s32[1]{0} %select.287, s32[] %select.396), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.53 (param_0.105: s32[75], param_1.679: s32[], param_2.808: s32[75]) -> s32[75] {
  %param_0.105 = s32[75]{0} parameter(0)
  %constant_4466 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.679 = s32[] parameter(1)
  %constant_4659 = s32[] constant(2)
  %add.1119 = s32[] add(s32[] %param_1.679, s32[] %constant_4659), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.880 = pred[] compare(s32[] %add.1119, s32[] %constant_4466), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4657 = s32[] constant(77)
  %add.1118 = s32[] add(s32[] %param_1.679, s32[] %constant_4657), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.388 = s32[] select(pred[] %compare.880, s32[] %add.1118, s32[] %add.1119), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.754 = pred[] compare(s32[] %constant_4466, s32[] %select.388), direction=LE
  %constant_4465 = s32[] constant(74)
  %compare.753 = pred[] compare(s32[] %constant_4465, s32[] %select.388), direction=GE
  %and.275 = pred[] and(pred[] %compare.754, pred[] %compare.753)
  %bitcast.121 = pred[1]{0} bitcast(pred[] %and.275)
  %param_2.808 = s32[75]{0} parameter(2)
  %dynamic-slice.632 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.808, s32[] %select.388), dynamic_slice_sizes={1}
  %bitcast.171 = s32[] bitcast(s32[1]{0} %dynamic-slice.632), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.883 = pred[] compare(s32[] %bitcast.171, s32[] %constant_4466), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4666 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1123 = s32[] add(s32[] %bitcast.171, s32[] %constant_4666), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.391 = s32[] select(pred[] %compare.883, s32[] %add.1123, s32[] %bitcast.171), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.579 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.105, s32[] %select.391), dynamic_slice_sizes={1}
  %dynamic-slice.764 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.105, s32[] %select.388), dynamic_slice_sizes={1}
  %select.288 = s32[1]{0} select(pred[1]{0} %bitcast.121, s32[1]{0} %dynamic-slice.579, s32[1]{0} %dynamic-slice.764)
  ROOT %dynamic-update-slice.248 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.105, s32[1]{0} %select.288, s32[] %select.388), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.56 (param_0.112: s32[75], param_1.677: s32[75], param_2.807: s32[], param_3.808: s32[75]) -> s32[75] {
  %param_0.112 = s32[75]{0} parameter(0)
  %constant_4471 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.677 = s32[75]{0} parameter(1)
  %param_2.807 = s32[] parameter(2)
  %constant_4638 = s32[] constant(1), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %add.1105 = s32[] add(s32[] %param_2.807, s32[] %constant_4638), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.874 = pred[] compare(s32[] %add.1105, s32[] %constant_4471), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4636 = s32[] constant(76)
  %add.1104 = s32[] add(s32[] %param_2.807, s32[] %constant_4636), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.381 = s32[] select(pred[] %compare.874, s32[] %add.1104, s32[] %add.1105), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.629 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.677, s32[] %select.381), dynamic_slice_sizes={1}
  %bitcast.169 = s32[] bitcast(s32[1]{0} %dynamic-slice.629), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.872 = pred[] compare(s32[] %bitcast.169, s32[] %constant_4471), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4634 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1103 = s32[] add(s32[] %bitcast.169, s32[] %constant_4634), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.380 = s32[] select(pred[] %compare.872, s32[] %add.1103, s32[] %bitcast.169), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.758 = pred[] compare(s32[] %constant_4471, s32[] %select.380), direction=LE
  %constant_4470 = s32[] constant(74)
  %compare.757 = pred[] compare(s32[] %constant_4470, s32[] %select.380), direction=GE
  %and.276 = pred[] and(pred[] %compare.758, pred[] %compare.757)
  %bitcast.123 = pred[1]{0} bitcast(pred[] %and.276)
  %param_3.808 = s32[75]{0} parameter(3)
  %dynamic-slice.762 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.808, s32[] %select.381), dynamic_slice_sizes={1}
  %dynamic-slice.581 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.112, s32[] %select.380), dynamic_slice_sizes={1}
  %select.292 = s32[1]{0} select(pred[1]{0} %bitcast.123, s32[1]{0} %dynamic-slice.762, s32[1]{0} %dynamic-slice.581)
  ROOT %dynamic-update-slice.249 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.112, s32[1]{0} %select.292, s32[] %select.380), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.57 (param_0.113: s32[75], param_1.675: s32[], param_2.806: s32[75]) -> s32[75] {
  %param_0.113 = s32[75]{0} parameter(0)
  %constant_4473 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.675 = s32[] parameter(1)
  %constant_4616 = s32[] constant(1), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %add.1091 = s32[] add(s32[] %param_1.675, s32[] %constant_4616), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.865 = pred[] compare(s32[] %add.1091, s32[] %constant_4473), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4614 = s32[] constant(76)
  %add.1090 = s32[] add(s32[] %param_1.675, s32[] %constant_4614), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.372 = s32[] select(pred[] %compare.865, s32[] %add.1090, s32[] %add.1091), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.761 = pred[] compare(s32[] %constant_4473, s32[] %select.372), direction=LE
  %constant_4472 = s32[] constant(74)
  %compare.760 = pred[] compare(s32[] %constant_4472, s32[] %select.372), direction=GE
  %and.277 = pred[] and(pred[] %compare.761, pred[] %compare.760)
  %bitcast.124 = pred[1]{0} bitcast(pred[] %and.277)
  %param_2.806 = s32[75]{0} parameter(2)
  %dynamic-slice.627 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.806, s32[] %select.372), dynamic_slice_sizes={1}
  %bitcast.167 = s32[] bitcast(s32[1]{0} %dynamic-slice.627), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.868 = pred[] compare(s32[] %bitcast.167, s32[] %constant_4473), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_4623 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1095 = s32[] add(s32[] %bitcast.167, s32[] %constant_4623), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.375 = s32[] select(pred[] %compare.868, s32[] %add.1095, s32[] %bitcast.167), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.582 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.113, s32[] %select.375), dynamic_slice_sizes={1}
  %dynamic-slice.760 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.113, s32[] %select.372), dynamic_slice_sizes={1}
  %select.293 = s32[1]{0} select(pred[1]{0} %bitcast.124, s32[1]{0} %dynamic-slice.582, s32[1]{0} %dynamic-slice.760)
  ROOT %dynamic-update-slice.250 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.113, s32[1]{0} %select.293, s32[] %select.372), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.60 (param_0.120: s32[75], param_1.673: s32[75], param_2.805: s32[], param_3.803: s32[75]) -> s32[75] {
  %param_0.120 = s32[75]{0} parameter(0)
  %constant_4478 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.673 = s32[75]{0} parameter(1)
  %param_2.805 = s32[] parameter(2)
  %compare.859 = pred[] compare(s32[] %param_2.805, s32[] %constant_4478), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4592 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1078 = s32[] add(s32[] %param_2.805, s32[] %constant_4592), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.365 = s32[] select(pred[] %compare.859, s32[] %add.1078, s32[] %param_2.805), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.624 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.673, s32[] %select.365), dynamic_slice_sizes={1}
  %bitcast.165 = s32[] bitcast(s32[1]{0} %dynamic-slice.624), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.858 = pred[] compare(s32[] %bitcast.165, s32[] %constant_4478), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %add.1077 = s32[] add(s32[] %bitcast.165, s32[] %constant_4592), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.364 = s32[] select(pred[] %compare.858, s32[] %add.1077, s32[] %bitcast.165), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.765 = pred[] compare(s32[] %constant_4478, s32[] %select.364), direction=LE
  %constant_4477 = s32[] constant(74)
  %compare.764 = pred[] compare(s32[] %constant_4477, s32[] %select.364), direction=GE
  %and.278 = pred[] and(pred[] %compare.765, pred[] %compare.764)
  %bitcast.126 = pred[1]{0} bitcast(pred[] %and.278)
  %param_3.803 = s32[75]{0} parameter(3)
  %dynamic-slice.758 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.803, s32[] %select.365), dynamic_slice_sizes={1}
  %dynamic-slice.584 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.120, s32[] %select.364), dynamic_slice_sizes={1}
  %select.296 = s32[1]{0} select(pred[1]{0} %bitcast.126, s32[1]{0} %dynamic-slice.758, s32[1]{0} %dynamic-slice.584)
  ROOT %dynamic-update-slice.251 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.120, s32[1]{0} %select.296, s32[] %select.364), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.61 (param_0.121: s32[75], param_1.671: s32[], param_2.804: s32[75]) -> s32[75] {
  %param_0.121 = s32[75]{0} parameter(0)
  %constant_4480 = s32[] constant(0), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %param_1.671 = s32[] parameter(1)
  %compare.851 = pred[] compare(s32[] %param_1.671, s32[] %constant_4480), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_4576 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1069 = s32[] add(s32[] %param_1.671, s32[] %constant_4576), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.356 = s32[] select(pred[] %compare.851, s32[] %add.1069, s32[] %param_1.671), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.767 = pred[] compare(s32[] %constant_4480, s32[] %select.356), direction=LE
  %constant_4479 = s32[] constant(74)
  %compare.766 = pred[] compare(s32[] %constant_4479, s32[] %select.356), direction=GE
  %and.279 = pred[] and(pred[] %compare.767, pred[] %compare.766)
  %bitcast.127 = pred[1]{0} bitcast(pred[] %and.279)
  %param_2.804 = s32[75]{0} parameter(2)
  %dynamic-slice.622 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.804, s32[] %select.356), dynamic_slice_sizes={1}
  %bitcast.163 = s32[] bitcast(s32[1]{0} %dynamic-slice.622), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/while/body/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                                      slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.854 = pred[] compare(s32[] %bitcast.163, s32[] %constant_4480), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/body/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %add.1073 = s32[] add(s32[] %bitcast.163, s32[] %constant_4576), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.360 = s32[] select(pred[] %compare.854, s32[] %add.1073, s32[] %bitcast.163), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/while/body/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.585 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.121, s32[] %select.360), dynamic_slice_sizes={1}
  %dynamic-slice.756 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.121, s32[] %select.356), dynamic_slice_sizes={1}
  %select.298 = s32[1]{0} select(pred[1]{0} %bitcast.127, s32[1]{0} %dynamic-slice.585, s32[1]{0} %dynamic-slice.756)
  ROOT %dynamic-update-slice.252 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.121, s32[1]{0} %select.298, s32[] %select.356), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%body_computation.141.clone (parameter.1: (s64[], s32[], s32[75], s32[75])) -> (s64[], s32[], s32[75], s32[75]) {
  %parameter.1 = (s64[], s32[], s32[75]{0}, s32[75]{0}) parameter(0)
  %get-tuple-element.2107 = s64[] get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %parameter.1), index=0
  %constant_2876 = s64[] constant(1), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  %add.510 = s64[] add(s64[] %get-tuple-element.2107, s64[] %constant_2876), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  %get-tuple-element.2109 = s32[75]{0} get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %parameter.1), index=2
  %copy = s32[75]{0} copy(s32[75]{0} %get-tuple-element.2109)
  %get-tuple-element.2108 = s32[] get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %parameter.1), index=1
  %get-tuple-element.2110 = s32[75]{0} get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %parameter.1), index=3
  %fusion.61 = s32[75]{0} fusion(s32[75]{0} %copy, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.61, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.60 = s32[75]{0} fusion(s32[75]{0} %fusion.61, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2109), kind=kLoop, calls=%fused_computation.60, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.2 = s32[75]{0} copy(s32[75]{0} %fusion.60)
  %fusion.57 = s32[75]{0} fusion(s32[75]{0} %copy.2, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.57, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.56 = s32[75]{0} fusion(s32[75]{0} %fusion.57, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.60), kind=kLoop, calls=%fused_computation.56, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.4 = s32[75]{0} copy(s32[75]{0} %fusion.56)
  %fusion.53 = s32[75]{0} fusion(s32[75]{0} %copy.4, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.53, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.52 = s32[75]{0} fusion(s32[75]{0} %fusion.53, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.56), kind=kLoop, calls=%fused_computation.52, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.6 = s32[75]{0} copy(s32[75]{0} %fusion.52)
  %fusion.49 = s32[75]{0} fusion(s32[75]{0} %copy.6, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.49, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.48 = s32[75]{0} fusion(s32[75]{0} %fusion.49, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.52), kind=kLoop, calls=%fused_computation.48, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.8 = s32[75]{0} copy(s32[75]{0} %fusion.48)
  %fusion.45 = s32[75]{0} fusion(s32[75]{0} %copy.8, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.45, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.44 = s32[75]{0} fusion(s32[75]{0} %fusion.45, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.48), kind=kLoop, calls=%fused_computation.44, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.10 = s32[75]{0} copy(s32[75]{0} %fusion.44)
  %fusion.41 = s32[75]{0} fusion(s32[75]{0} %copy.10, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.41, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.40 = s32[75]{0} fusion(s32[75]{0} %fusion.41, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.44), kind=kLoop, calls=%fused_computation.40, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.12 = s32[75]{0} copy(s32[75]{0} %fusion.40)
  %fusion.37 = s32[75]{0} fusion(s32[75]{0} %copy.12, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.37, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.36 = s32[75]{0} fusion(s32[75]{0} %fusion.37, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.40), kind=kLoop, calls=%fused_computation.36, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.14 = s32[75]{0} copy(s32[75]{0} %fusion.36)
  %fusion.33 = s32[75]{0} fusion(s32[75]{0} %copy.14, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.33, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.32 = s32[75]{0} fusion(s32[75]{0} %fusion.33, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.36), kind=kLoop, calls=%fused_computation.32, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.16 = s32[75]{0} copy(s32[75]{0} %fusion.32)
  %fusion.29 = s32[75]{0} fusion(s32[75]{0} %copy.16, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.29, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.28 = s32[75]{0} fusion(s32[75]{0} %fusion.29, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.32), kind=kLoop, calls=%fused_computation.28, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.18 = s32[75]{0} copy(s32[75]{0} %fusion.28)
  %fusion.25 = s32[75]{0} fusion(s32[75]{0} %copy.18, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.25, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.24 = s32[75]{0} fusion(s32[75]{0} %fusion.25, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.28), kind=kLoop, calls=%fused_computation.24, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.20 = s32[75]{0} copy(s32[75]{0} %fusion.24)
  %fusion.21 = s32[75]{0} fusion(s32[75]{0} %copy.20, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.21, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.20 = s32[75]{0} fusion(s32[75]{0} %fusion.21, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.24), kind=kLoop, calls=%fused_computation.20, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.22 = s32[75]{0} copy(s32[75]{0} %fusion.20)
  %fusion.17 = s32[75]{0} fusion(s32[75]{0} %copy.22, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.17, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.16 = s32[75]{0} fusion(s32[75]{0} %fusion.17, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.20), kind=kLoop, calls=%fused_computation.16, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.24 = s32[75]{0} copy(s32[75]{0} %fusion.16)
  %fusion.13 = s32[75]{0} fusion(s32[75]{0} %copy.24, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.13, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.12 = s32[75]{0} fusion(s32[75]{0} %fusion.13, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.16), kind=kLoop, calls=%fused_computation.12, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.26 = s32[75]{0} copy(s32[75]{0} %fusion.12)
  %fusion.9 = s32[75]{0} fusion(s32[75]{0} %copy.26, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.9, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.8 = s32[75]{0} fusion(s32[75]{0} %fusion.9, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.12), kind=kLoop, calls=%fused_computation.8, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.28 = s32[75]{0} copy(s32[75]{0} %fusion.8)
  %fusion.5 = s32[75]{0} fusion(s32[75]{0} %copy.28, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.5, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.4 = s32[75]{0} fusion(s32[75]{0} %fusion.5, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.8), kind=kLoop, calls=%fused_computation.4, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.30 = s32[75]{0} copy(s32[75]{0} %fusion.4)
  %fusion.1 = s32[75]{0} fusion(s32[75]{0} %copy.30, s32[] %get-tuple-element.2108, s32[75]{0} %get-tuple-element.2110), kind=kLoop, calls=%fused_computation.1, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion = s32[75]{0} fusion(s32[75]{0} %fusion.1, s32[75]{0} %get-tuple-element.2110, s32[] %get-tuple-element.2108, s32[75]{0} %fusion.4), kind=kLoop, calls=%fused_computation, control-predecessors={%fusion.60, %fusion.61}, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/while/body/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                                       indices_are_sorted=False\n                                                       unique_indices=False\n                                                       update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %constant_4279 = s32[] constant(16)
  %add.879 = s32[] add(s32[] %get-tuple-element.2108, s32[] %constant_4279), control-predecessors={%fusion.13, %fusion.33, %fusion.1, %fusion.25, %fusion.17, %fusion.21, %fusion.29, %fusion.37, %fusion.61, %fusion.49, %fusion.5, %fusion.41, %fusion.57, %fusion.9, %fusion.45, %fusion.53, %fusion.60, %fusion.28, %fusion.56, %fusion, %fusion.52, %fusion.20, %fusion.48, %fusion.4, %fusion.44, %fusion.24, %fusion.40, %fusion.8, %fusion.36, %fusion.16, %fusion.32, %fusion.12}, metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/while/body/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  ROOT %tuple.923 = (s64[], s32[], s32[75]{0}, s32[75]{0}) tuple(s64[] %add.510, s32[] %add.879, s32[75]{0} %fusion, s32[75]{0} %get-tuple-element.2110)
}

%cond_computation.860.clone (parameter.0: (s64[], s32[], s32[75], s32[75])) -> pred[] {
  %parameter.0 = (s64[], s32[], s32[75]{0}, s32[75]{0}) parameter(0)
  %get-tuple-element.1387 = s64[] get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %parameter.0), index=0
  %constant_2651 = s64[] constant(4), metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/cond/lt" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  ROOT %compare.327 = pred[] compare(s64[] %get-tuple-element.1387, s64[] %constant_2651), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/while/cond/lt" source_file="<ipython-input-6-f97d46683450>" source_line=1}
}

%fused_computation.64 (param_0.127: s32[75], param_1.777: s32[75], param_2.857: s32[], param_3.932: s32[75]) -> s32[75] {
  %param_0.127 = s32[75]{0} parameter(0)
  %constant_4486 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.777 = s32[75]{0} parameter(1)
  %param_2.857 = s32[] parameter(2)
  %constant_5703 = s32[] constant(10)
  %add.1736 = s32[] add(s32[] %param_2.857, s32[] %constant_5703), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1260 = pred[] compare(s32[] %add.1736, s32[] %constant_4486), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5701 = s32[] constant(85)
  %add.1735 = s32[] add(s32[] %param_2.857, s32[] %constant_5701), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.776 = s32[] select(pred[] %compare.1260, s32[] %add.1735, s32[] %add.1736), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.754 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.777, s32[] %select.776), dynamic_slice_sizes={1}
  %bitcast.269 = s32[] bitcast(s32[1]{0} %dynamic-slice.754), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1259 = pred[] compare(s32[] %bitcast.269, s32[] %constant_4486), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5699 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1734 = s32[] add(s32[] %bitcast.269, s32[] %constant_5699), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.775 = s32[] select(pred[] %compare.1259, s32[] %add.1734, s32[] %bitcast.269), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.772 = pred[] compare(s32[] %constant_4486, s32[] %select.775), direction=LE
  %constant_4485 = s32[] constant(74)
  %compare.771 = pred[] compare(s32[] %constant_4485, s32[] %select.775), direction=GE
  %and.280 = pred[] and(pred[] %compare.772, pred[] %compare.771)
  %bitcast.129 = pred[1]{0} bitcast(pred[] %and.280)
  %param_3.932 = s32[75]{0} parameter(3)
  %dynamic-slice.862 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.932, s32[] %select.776), dynamic_slice_sizes={1}
  %dynamic-slice.587 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.127, s32[] %select.775), dynamic_slice_sizes={1}
  %select.301 = s32[1]{0} select(pred[1]{0} %bitcast.129, s32[1]{0} %dynamic-slice.862, s32[1]{0} %dynamic-slice.587)
  ROOT %dynamic-update-slice.253 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.127, s32[1]{0} %select.301, s32[] %select.775), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.65 (param_0.128: s32[75], param_1.775: s32[], param_2.856: s32[75]) -> s32[75] {
  %param_0.128 = s32[75]{0} parameter(0)
  %constant_4488 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.775 = s32[] parameter(1)
  %constant_5681 = s32[] constant(10)
  %add.1724 = s32[] add(s32[] %param_1.775, s32[] %constant_5681), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1251 = pred[] compare(s32[] %add.1724, s32[] %constant_4488), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5679 = s32[] constant(85)
  %add.1723 = s32[] add(s32[] %param_1.775, s32[] %constant_5679), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.767 = s32[] select(pred[] %compare.1251, s32[] %add.1723, s32[] %add.1724), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.774 = pred[] compare(s32[] %constant_4488, s32[] %select.767), direction=LE
  %constant_4487 = s32[] constant(74)
  %compare.773 = pred[] compare(s32[] %constant_4487, s32[] %select.767), direction=GE
  %and.281 = pred[] and(pred[] %compare.774, pred[] %compare.773)
  %bitcast.130 = pred[1]{0} bitcast(pred[] %and.281)
  %param_2.856 = s32[75]{0} parameter(2)
  %dynamic-slice.752 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.856, s32[] %select.767), dynamic_slice_sizes={1}
  %bitcast.267 = s32[] bitcast(s32[1]{0} %dynamic-slice.752), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1255 = pred[] compare(s32[] %bitcast.267, s32[] %constant_4488), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5688 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1728 = s32[] add(s32[] %bitcast.267, s32[] %constant_5688), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.770 = s32[] select(pred[] %compare.1255, s32[] %add.1728, s32[] %bitcast.267), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.588 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.128, s32[] %select.770), dynamic_slice_sizes={1}
  %dynamic-slice.860 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.128, s32[] %select.767), dynamic_slice_sizes={1}
  %select.302 = s32[1]{0} select(pred[1]{0} %bitcast.130, s32[1]{0} %dynamic-slice.588, s32[1]{0} %dynamic-slice.860)
  ROOT %dynamic-update-slice.254 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.128, s32[1]{0} %select.302, s32[] %select.767), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.68 (param_0.135: s32[75], param_1.773: s32[75], param_2.855: s32[], param_3.927: s32[75]) -> s32[75] {
  %param_0.135 = s32[75]{0} parameter(0)
  %constant_4493 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.773 = s32[75]{0} parameter(1)
  %param_2.855 = s32[] parameter(2)
  %constant_5660 = s32[] constant(9)
  %add.1712 = s32[] add(s32[] %param_2.855, s32[] %constant_5660), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1245 = pred[] compare(s32[] %add.1712, s32[] %constant_4493), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5658 = s32[] constant(84)
  %add.1711 = s32[] add(s32[] %param_2.855, s32[] %constant_5658), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.760 = s32[] select(pred[] %compare.1245, s32[] %add.1711, s32[] %add.1712), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.749 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.773, s32[] %select.760), dynamic_slice_sizes={1}
  %bitcast.265 = s32[] bitcast(s32[1]{0} %dynamic-slice.749), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1244 = pred[] compare(s32[] %bitcast.265, s32[] %constant_4493), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5656 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1710 = s32[] add(s32[] %bitcast.265, s32[] %constant_5656), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.759 = s32[] select(pred[] %compare.1244, s32[] %add.1710, s32[] %bitcast.265), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.779 = pred[] compare(s32[] %constant_4493, s32[] %select.759), direction=LE
  %constant_4492 = s32[] constant(74)
  %compare.777 = pred[] compare(s32[] %constant_4492, s32[] %select.759), direction=GE
  %and.282 = pred[] and(pred[] %compare.779, pred[] %compare.777)
  %bitcast.132 = pred[1]{0} bitcast(pred[] %and.282)
  %param_3.927 = s32[75]{0} parameter(3)
  %dynamic-slice.858 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.927, s32[] %select.760), dynamic_slice_sizes={1}
  %dynamic-slice.590 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.135, s32[] %select.759), dynamic_slice_sizes={1}
  %select.306 = s32[1]{0} select(pred[1]{0} %bitcast.132, s32[1]{0} %dynamic-slice.858, s32[1]{0} %dynamic-slice.590)
  ROOT %dynamic-update-slice.255 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.135, s32[1]{0} %select.306, s32[] %select.759), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.69 (param_0.136: s32[75], param_1.771: s32[], param_2.854: s32[75]) -> s32[75] {
  %param_0.136 = s32[75]{0} parameter(0)
  %constant_4495 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.771 = s32[] parameter(1)
  %constant_5638 = s32[] constant(9)
  %add.1700 = s32[] add(s32[] %param_1.771, s32[] %constant_5638), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1236 = pred[] compare(s32[] %add.1700, s32[] %constant_4495), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5636 = s32[] constant(84)
  %add.1699 = s32[] add(s32[] %param_1.771, s32[] %constant_5636), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.751 = s32[] select(pred[] %compare.1236, s32[] %add.1699, s32[] %add.1700), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.781 = pred[] compare(s32[] %constant_4495, s32[] %select.751), direction=LE
  %constant_4494 = s32[] constant(74)
  %compare.780 = pred[] compare(s32[] %constant_4494, s32[] %select.751), direction=GE
  %and.283 = pred[] and(pred[] %compare.781, pred[] %compare.780)
  %bitcast.133 = pred[1]{0} bitcast(pred[] %and.283)
  %param_2.854 = s32[75]{0} parameter(2)
  %dynamic-slice.747 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.854, s32[] %select.751), dynamic_slice_sizes={1}
  %bitcast.263 = s32[] bitcast(s32[1]{0} %dynamic-slice.747), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1240 = pred[] compare(s32[] %bitcast.263, s32[] %constant_4495), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5645 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1704 = s32[] add(s32[] %bitcast.263, s32[] %constant_5645), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.755 = s32[] select(pred[] %compare.1240, s32[] %add.1704, s32[] %bitcast.263), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.591 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.136, s32[] %select.755), dynamic_slice_sizes={1}
  %dynamic-slice.856 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.136, s32[] %select.751), dynamic_slice_sizes={1}
  %select.307 = s32[1]{0} select(pred[1]{0} %bitcast.133, s32[1]{0} %dynamic-slice.591, s32[1]{0} %dynamic-slice.856)
  ROOT %dynamic-update-slice.256 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.136, s32[1]{0} %select.307, s32[] %select.751), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.72 (param_0.143: s32[75], param_1.769: s32[75], param_2.853: s32[], param_3.922: s32[75]) -> s32[75] {
  %param_0.143 = s32[75]{0} parameter(0)
  %constant_4500 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.769 = s32[75]{0} parameter(1)
  %param_2.853 = s32[] parameter(2)
  %constant_5617 = s32[] constant(8)
  %add.1688 = s32[] add(s32[] %param_2.853, s32[] %constant_5617), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1230 = pred[] compare(s32[] %add.1688, s32[] %constant_4500), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5615 = s32[] constant(83)
  %add.1687 = s32[] add(s32[] %param_2.853, s32[] %constant_5615), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.744 = s32[] select(pred[] %compare.1230, s32[] %add.1687, s32[] %add.1688), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.744 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.769, s32[] %select.744), dynamic_slice_sizes={1}
  %bitcast.261 = s32[] bitcast(s32[1]{0} %dynamic-slice.744), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1229 = pred[] compare(s32[] %bitcast.261, s32[] %constant_4500), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5613 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1686 = s32[] add(s32[] %bitcast.261, s32[] %constant_5613), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.743 = s32[] select(pred[] %compare.1229, s32[] %add.1686, s32[] %bitcast.261), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.785 = pred[] compare(s32[] %constant_4500, s32[] %select.743), direction=LE
  %constant_4499 = s32[] constant(74)
  %compare.784 = pred[] compare(s32[] %constant_4499, s32[] %select.743), direction=GE
  %and.284 = pred[] and(pred[] %compare.785, pred[] %compare.784)
  %bitcast.135 = pred[1]{0} bitcast(pred[] %and.284)
  %param_3.922 = s32[75]{0} parameter(3)
  %dynamic-slice.854 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.922, s32[] %select.744), dynamic_slice_sizes={1}
  %dynamic-slice.593 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.143, s32[] %select.743), dynamic_slice_sizes={1}
  %select.310 = s32[1]{0} select(pred[1]{0} %bitcast.135, s32[1]{0} %dynamic-slice.854, s32[1]{0} %dynamic-slice.593)
  ROOT %dynamic-update-slice.257 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.143, s32[1]{0} %select.310, s32[] %select.743), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.73 (param_0.144: s32[75], param_1.767: s32[], param_2.852: s32[75]) -> s32[75] {
  %param_0.144 = s32[75]{0} parameter(0)
  %constant_4502 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.767 = s32[] parameter(1)
  %constant_5595 = s32[] constant(8)
  %add.1676 = s32[] add(s32[] %param_1.767, s32[] %constant_5595), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1221 = pred[] compare(s32[] %add.1676, s32[] %constant_4502), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5593 = s32[] constant(83)
  %add.1675 = s32[] add(s32[] %param_1.767, s32[] %constant_5593), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.735 = s32[] select(pred[] %compare.1221, s32[] %add.1675, s32[] %add.1676), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.788 = pred[] compare(s32[] %constant_4502, s32[] %select.735), direction=LE
  %constant_4501 = s32[] constant(74)
  %compare.787 = pred[] compare(s32[] %constant_4501, s32[] %select.735), direction=GE
  %and.285 = pred[] and(pred[] %compare.788, pred[] %compare.787)
  %bitcast.136 = pred[1]{0} bitcast(pred[] %and.285)
  %param_2.852 = s32[75]{0} parameter(2)
  %dynamic-slice.742 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.852, s32[] %select.735), dynamic_slice_sizes={1}
  %bitcast.259 = s32[] bitcast(s32[1]{0} %dynamic-slice.742), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1224 = pred[] compare(s32[] %bitcast.259, s32[] %constant_4502), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5602 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1680 = s32[] add(s32[] %bitcast.259, s32[] %constant_5602), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.739 = s32[] select(pred[] %compare.1224, s32[] %add.1680, s32[] %bitcast.259), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.594 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.144, s32[] %select.739), dynamic_slice_sizes={1}
  %dynamic-slice.852 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.144, s32[] %select.735), dynamic_slice_sizes={1}
  %select.311 = s32[1]{0} select(pred[1]{0} %bitcast.136, s32[1]{0} %dynamic-slice.594, s32[1]{0} %dynamic-slice.852)
  ROOT %dynamic-update-slice.258 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.144, s32[1]{0} %select.311, s32[] %select.735), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.76 (param_0.151: s32[75], param_1.765: s32[75], param_2.851: s32[], param_3.917: s32[75]) -> s32[75] {
  %param_0.151 = s32[75]{0} parameter(0)
  %constant_4507 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.765 = s32[75]{0} parameter(1)
  %param_2.851 = s32[] parameter(2)
  %constant_5574 = s32[] constant(7)
  %add.1664 = s32[] add(s32[] %param_2.851, s32[] %constant_5574), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1214 = pred[] compare(s32[] %add.1664, s32[] %constant_4507), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5572 = s32[] constant(82)
  %add.1663 = s32[] add(s32[] %param_2.851, s32[] %constant_5572), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.728 = s32[] select(pred[] %compare.1214, s32[] %add.1663, s32[] %add.1664), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.739 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.765, s32[] %select.728), dynamic_slice_sizes={1}
  %bitcast.257 = s32[] bitcast(s32[1]{0} %dynamic-slice.739), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1213 = pred[] compare(s32[] %bitcast.257, s32[] %constant_4507), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5570 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1662 = s32[] add(s32[] %bitcast.257, s32[] %constant_5570), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.727 = s32[] select(pred[] %compare.1213, s32[] %add.1662, s32[] %bitcast.257), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.792 = pred[] compare(s32[] %constant_4507, s32[] %select.727), direction=LE
  %constant_4506 = s32[] constant(74)
  %compare.791 = pred[] compare(s32[] %constant_4506, s32[] %select.727), direction=GE
  %and.286 = pred[] and(pred[] %compare.792, pred[] %compare.791)
  %bitcast.138 = pred[1]{0} bitcast(pred[] %and.286)
  %param_3.917 = s32[75]{0} parameter(3)
  %dynamic-slice.850 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.917, s32[] %select.728), dynamic_slice_sizes={1}
  %dynamic-slice.596 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.151, s32[] %select.727), dynamic_slice_sizes={1}
  %select.315 = s32[1]{0} select(pred[1]{0} %bitcast.138, s32[1]{0} %dynamic-slice.850, s32[1]{0} %dynamic-slice.596)
  ROOT %dynamic-update-slice.259 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.151, s32[1]{0} %select.315, s32[] %select.727), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.77 (param_0.152: s32[75], param_1.763: s32[], param_2.850: s32[75]) -> s32[75] {
  %param_0.152 = s32[75]{0} parameter(0)
  %constant_4509 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.763 = s32[] parameter(1)
  %constant_5552 = s32[] constant(7)
  %add.1652 = s32[] add(s32[] %param_1.763, s32[] %constant_5552), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1205 = pred[] compare(s32[] %add.1652, s32[] %constant_4509), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5550 = s32[] constant(82)
  %add.1651 = s32[] add(s32[] %param_1.763, s32[] %constant_5550), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.720 = s32[] select(pred[] %compare.1205, s32[] %add.1651, s32[] %add.1652), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.795 = pred[] compare(s32[] %constant_4509, s32[] %select.720), direction=LE
  %constant_4508 = s32[] constant(74)
  %compare.793 = pred[] compare(s32[] %constant_4508, s32[] %select.720), direction=GE
  %and.287 = pred[] and(pred[] %compare.795, pred[] %compare.793)
  %bitcast.139 = pred[1]{0} bitcast(pred[] %and.287)
  %param_2.850 = s32[75]{0} parameter(2)
  %dynamic-slice.737 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.850, s32[] %select.720), dynamic_slice_sizes={1}
  %bitcast.255 = s32[] bitcast(s32[1]{0} %dynamic-slice.737), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1208 = pred[] compare(s32[] %bitcast.255, s32[] %constant_4509), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5559 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1656 = s32[] add(s32[] %bitcast.255, s32[] %constant_5559), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.723 = s32[] select(pred[] %compare.1208, s32[] %add.1656, s32[] %bitcast.255), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.597 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.152, s32[] %select.723), dynamic_slice_sizes={1}
  %dynamic-slice.848 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.152, s32[] %select.720), dynamic_slice_sizes={1}
  %select.316 = s32[1]{0} select(pred[1]{0} %bitcast.139, s32[1]{0} %dynamic-slice.597, s32[1]{0} %dynamic-slice.848)
  ROOT %dynamic-update-slice.260 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.152, s32[1]{0} %select.316, s32[] %select.720), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.80 (param_0.159: s32[75], param_1.761: s32[75], param_2.849: s32[], param_3.912: s32[75]) -> s32[75] {
  %param_0.159 = s32[75]{0} parameter(0)
  %constant_4514 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.761 = s32[75]{0} parameter(1)
  %param_2.849 = s32[] parameter(2)
  %constant_5531 = s32[] constant(6)
  %add.1640 = s32[] add(s32[] %param_2.849, s32[] %constant_5531), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1198 = pred[] compare(s32[] %add.1640, s32[] %constant_4514), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5529 = s32[] constant(81)
  %add.1639 = s32[] add(s32[] %param_2.849, s32[] %constant_5529), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.713 = s32[] select(pred[] %compare.1198, s32[] %add.1639, s32[] %add.1640), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.734 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.761, s32[] %select.713), dynamic_slice_sizes={1}
  %bitcast.253 = s32[] bitcast(s32[1]{0} %dynamic-slice.734), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1197 = pred[] compare(s32[] %bitcast.253, s32[] %constant_4514), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5527 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1638 = s32[] add(s32[] %bitcast.253, s32[] %constant_5527), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.712 = s32[] select(pred[] %compare.1197, s32[] %add.1638, s32[] %bitcast.253), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.799 = pred[] compare(s32[] %constant_4514, s32[] %select.712), direction=LE
  %constant_4513 = s32[] constant(74)
  %compare.798 = pred[] compare(s32[] %constant_4513, s32[] %select.712), direction=GE
  %and.288 = pred[] and(pred[] %compare.799, pred[] %compare.798)
  %bitcast.141 = pred[1]{0} bitcast(pred[] %and.288)
  %param_3.912 = s32[75]{0} parameter(3)
  %dynamic-slice.846 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.912, s32[] %select.713), dynamic_slice_sizes={1}
  %dynamic-slice.599 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.159, s32[] %select.712), dynamic_slice_sizes={1}
  %select.319 = s32[1]{0} select(pred[1]{0} %bitcast.141, s32[1]{0} %dynamic-slice.846, s32[1]{0} %dynamic-slice.599)
  ROOT %dynamic-update-slice.261 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.159, s32[1]{0} %select.319, s32[] %select.712), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.81 (param_0.160: s32[75], param_1.759: s32[], param_2.848: s32[75]) -> s32[75] {
  %param_0.160 = s32[75]{0} parameter(0)
  %constant_4516 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.759 = s32[] parameter(1)
  %constant_5509 = s32[] constant(6)
  %add.1628 = s32[] add(s32[] %param_1.759, s32[] %constant_5509), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1190 = pred[] compare(s32[] %add.1628, s32[] %constant_4516), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5507 = s32[] constant(81)
  %add.1627 = s32[] add(s32[] %param_1.759, s32[] %constant_5507), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.704 = s32[] select(pred[] %compare.1190, s32[] %add.1627, s32[] %add.1628), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.801 = pred[] compare(s32[] %constant_4516, s32[] %select.704), direction=LE
  %constant_4515 = s32[] constant(74)
  %compare.800 = pred[] compare(s32[] %constant_4515, s32[] %select.704), direction=GE
  %and.289 = pred[] and(pred[] %compare.801, pred[] %compare.800)
  %bitcast.142 = pred[1]{0} bitcast(pred[] %and.289)
  %param_2.848 = s32[75]{0} parameter(2)
  %dynamic-slice.732 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.848, s32[] %select.704), dynamic_slice_sizes={1}
  %bitcast.251 = s32[] bitcast(s32[1]{0} %dynamic-slice.732), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1193 = pred[] compare(s32[] %bitcast.251, s32[] %constant_4516), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5516 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1632 = s32[] add(s32[] %bitcast.251, s32[] %constant_5516), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.707 = s32[] select(pred[] %compare.1193, s32[] %add.1632, s32[] %bitcast.251), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.600 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.160, s32[] %select.707), dynamic_slice_sizes={1}
  %dynamic-slice.844 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.160, s32[] %select.704), dynamic_slice_sizes={1}
  %select.320 = s32[1]{0} select(pred[1]{0} %bitcast.142, s32[1]{0} %dynamic-slice.600, s32[1]{0} %dynamic-slice.844)
  ROOT %dynamic-update-slice.262 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.160, s32[1]{0} %select.320, s32[] %select.704), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.84 (param_0.167: s32[75], param_1.757: s32[75], param_2.847: s32[], param_3.907: s32[75]) -> s32[75] {
  %param_0.167 = s32[75]{0} parameter(0)
  %constant_4521 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.757 = s32[75]{0} parameter(1)
  %param_2.847 = s32[] parameter(2)
  %constant_5488 = s32[] constant(5)
  %add.1616 = s32[] add(s32[] %param_2.847, s32[] %constant_5488), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1183 = pred[] compare(s32[] %add.1616, s32[] %constant_4521), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5486 = s32[] constant(80)
  %add.1615 = s32[] add(s32[] %param_2.847, s32[] %constant_5486), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.697 = s32[] select(pred[] %compare.1183, s32[] %add.1615, s32[] %add.1616), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.729 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.757, s32[] %select.697), dynamic_slice_sizes={1}
  %bitcast.249 = s32[] bitcast(s32[1]{0} %dynamic-slice.729), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1182 = pred[] compare(s32[] %bitcast.249, s32[] %constant_4521), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5484 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1614 = s32[] add(s32[] %bitcast.249, s32[] %constant_5484), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.696 = s32[] select(pred[] %compare.1182, s32[] %add.1614, s32[] %bitcast.249), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.806 = pred[] compare(s32[] %constant_4521, s32[] %select.696), direction=LE
  %constant_4520 = s32[] constant(74)
  %compare.805 = pred[] compare(s32[] %constant_4520, s32[] %select.696), direction=GE
  %and.290 = pred[] and(pred[] %compare.806, pred[] %compare.805)
  %bitcast.144 = pred[1]{0} bitcast(pred[] %and.290)
  %param_3.907 = s32[75]{0} parameter(3)
  %dynamic-slice.842 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.907, s32[] %select.697), dynamic_slice_sizes={1}
  %dynamic-slice.602 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.167, s32[] %select.696), dynamic_slice_sizes={1}
  %select.324 = s32[1]{0} select(pred[1]{0} %bitcast.144, s32[1]{0} %dynamic-slice.842, s32[1]{0} %dynamic-slice.602)
  ROOT %dynamic-update-slice.263 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.167, s32[1]{0} %select.324, s32[] %select.696), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.85 (param_0.168: s32[75], param_1.755: s32[], param_2.846: s32[75]) -> s32[75] {
  %param_0.168 = s32[75]{0} parameter(0)
  %constant_4523 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.755 = s32[] parameter(1)
  %constant_5466 = s32[] constant(5)
  %add.1604 = s32[] add(s32[] %param_1.755, s32[] %constant_5466), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1175 = pred[] compare(s32[] %add.1604, s32[] %constant_4523), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5464 = s32[] constant(80)
  %add.1603 = s32[] add(s32[] %param_1.755, s32[] %constant_5464), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.688 = s32[] select(pred[] %compare.1175, s32[] %add.1603, s32[] %add.1604), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.808 = pred[] compare(s32[] %constant_4523, s32[] %select.688), direction=LE
  %constant_4522 = s32[] constant(74)
  %compare.807 = pred[] compare(s32[] %constant_4522, s32[] %select.688), direction=GE
  %and.291 = pred[] and(pred[] %compare.808, pred[] %compare.807)
  %bitcast.145 = pred[1]{0} bitcast(pred[] %and.291)
  %param_2.846 = s32[75]{0} parameter(2)
  %dynamic-slice.727 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.846, s32[] %select.688), dynamic_slice_sizes={1}
  %bitcast.247 = s32[] bitcast(s32[1]{0} %dynamic-slice.727), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1178 = pred[] compare(s32[] %bitcast.247, s32[] %constant_4523), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5473 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1608 = s32[] add(s32[] %bitcast.247, s32[] %constant_5473), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.691 = s32[] select(pred[] %compare.1178, s32[] %add.1608, s32[] %bitcast.247), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.603 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.168, s32[] %select.691), dynamic_slice_sizes={1}
  %dynamic-slice.840 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.168, s32[] %select.688), dynamic_slice_sizes={1}
  %select.325 = s32[1]{0} select(pred[1]{0} %bitcast.145, s32[1]{0} %dynamic-slice.603, s32[1]{0} %dynamic-slice.840)
  ROOT %dynamic-update-slice.264 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.168, s32[1]{0} %select.325, s32[] %select.688), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.88 (param_0.175: s32[75], param_1.753: s32[75], param_2.845: s32[], param_3.902: s32[75]) -> s32[75] {
  %param_0.175 = s32[75]{0} parameter(0)
  %constant_4528 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.753 = s32[75]{0} parameter(1)
  %param_2.845 = s32[] parameter(2)
  %constant_5445 = s32[] constant(4)
  %add.1592 = s32[] add(s32[] %param_2.845, s32[] %constant_5445), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1168 = pred[] compare(s32[] %add.1592, s32[] %constant_4528), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5443 = s32[] constant(79)
  %add.1591 = s32[] add(s32[] %param_2.845, s32[] %constant_5443), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.681 = s32[] select(pred[] %compare.1168, s32[] %add.1591, s32[] %add.1592), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.724 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.753, s32[] %select.681), dynamic_slice_sizes={1}
  %bitcast.245 = s32[] bitcast(s32[1]{0} %dynamic-slice.724), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1167 = pred[] compare(s32[] %bitcast.245, s32[] %constant_4528), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5441 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1590 = s32[] add(s32[] %bitcast.245, s32[] %constant_5441), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.680 = s32[] select(pred[] %compare.1167, s32[] %add.1590, s32[] %bitcast.245), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.812 = pred[] compare(s32[] %constant_4528, s32[] %select.680), direction=LE
  %constant_4527 = s32[] constant(74)
  %compare.811 = pred[] compare(s32[] %constant_4527, s32[] %select.680), direction=GE
  %and.292 = pred[] and(pred[] %compare.812, pred[] %compare.811)
  %bitcast.147 = pred[1]{0} bitcast(pred[] %and.292)
  %param_3.902 = s32[75]{0} parameter(3)
  %dynamic-slice.838 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.902, s32[] %select.681), dynamic_slice_sizes={1}
  %dynamic-slice.605 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.175, s32[] %select.680), dynamic_slice_sizes={1}
  %select.328 = s32[1]{0} select(pred[1]{0} %bitcast.147, s32[1]{0} %dynamic-slice.838, s32[1]{0} %dynamic-slice.605)
  ROOT %dynamic-update-slice.265 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.175, s32[1]{0} %select.328, s32[] %select.680), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.89 (param_0.176: s32[75], param_1.751: s32[], param_2.844: s32[75]) -> s32[75] {
  %param_0.176 = s32[75]{0} parameter(0)
  %constant_4530 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.751 = s32[] parameter(1)
  %constant_5423 = s32[] constant(4)
  %add.1580 = s32[] add(s32[] %param_1.751, s32[] %constant_5423), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1159 = pred[] compare(s32[] %add.1580, s32[] %constant_4530), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5421 = s32[] constant(79)
  %add.1579 = s32[] add(s32[] %param_1.751, s32[] %constant_5421), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.672 = s32[] select(pred[] %compare.1159, s32[] %add.1579, s32[] %add.1580), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.815 = pred[] compare(s32[] %constant_4530, s32[] %select.672), direction=LE
  %constant_4529 = s32[] constant(74)
  %compare.813 = pred[] compare(s32[] %constant_4529, s32[] %select.672), direction=GE
  %and.293 = pred[] and(pred[] %compare.815, pred[] %compare.813)
  %bitcast.148 = pred[1]{0} bitcast(pred[] %and.293)
  %param_2.844 = s32[75]{0} parameter(2)
  %dynamic-slice.722 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.844, s32[] %select.672), dynamic_slice_sizes={1}
  %bitcast.243 = s32[] bitcast(s32[1]{0} %dynamic-slice.722), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1162 = pred[] compare(s32[] %bitcast.243, s32[] %constant_4530), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5430 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1584 = s32[] add(s32[] %bitcast.243, s32[] %constant_5430), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.676 = s32[] select(pred[] %compare.1162, s32[] %add.1584, s32[] %bitcast.243), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.606 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.176, s32[] %select.676), dynamic_slice_sizes={1}
  %dynamic-slice.836 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.176, s32[] %select.672), dynamic_slice_sizes={1}
  %select.329 = s32[1]{0} select(pred[1]{0} %bitcast.148, s32[1]{0} %dynamic-slice.606, s32[1]{0} %dynamic-slice.836)
  ROOT %dynamic-update-slice.266 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.176, s32[1]{0} %select.329, s32[] %select.672), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.92 (param_0.183: s32[75], param_1.749: s32[75], param_2.843: s32[], param_3.897: s32[75]) -> s32[75] {
  %param_0.183 = s32[75]{0} parameter(0)
  %constant_4535 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.749 = s32[75]{0} parameter(1)
  %param_2.843 = s32[] parameter(2)
  %constant_5402 = s32[] constant(3)
  %add.1568 = s32[] add(s32[] %param_2.843, s32[] %constant_5402), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1152 = pred[] compare(s32[] %add.1568, s32[] %constant_4535), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5400 = s32[] constant(78)
  %add.1567 = s32[] add(s32[] %param_2.843, s32[] %constant_5400), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.666 = s32[] select(pred[] %compare.1152, s32[] %add.1567, s32[] %add.1568), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.719 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.749, s32[] %select.666), dynamic_slice_sizes={1}
  %bitcast.241 = s32[] bitcast(s32[1]{0} %dynamic-slice.719), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1151 = pred[] compare(s32[] %bitcast.241, s32[] %constant_4535), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5398 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1566 = s32[] add(s32[] %bitcast.241, s32[] %constant_5398), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.664 = s32[] select(pred[] %compare.1151, s32[] %add.1566, s32[] %bitcast.241), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.819 = pred[] compare(s32[] %constant_4535, s32[] %select.664), direction=LE
  %constant_4534 = s32[] constant(74)
  %compare.818 = pred[] compare(s32[] %constant_4534, s32[] %select.664), direction=GE
  %and.294 = pred[] and(pred[] %compare.819, pred[] %compare.818)
  %bitcast.150 = pred[1]{0} bitcast(pred[] %and.294)
  %param_3.897 = s32[75]{0} parameter(3)
  %dynamic-slice.834 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.897, s32[] %select.666), dynamic_slice_sizes={1}
  %dynamic-slice.608 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.183, s32[] %select.664), dynamic_slice_sizes={1}
  %select.332 = s32[1]{0} select(pred[1]{0} %bitcast.150, s32[1]{0} %dynamic-slice.834, s32[1]{0} %dynamic-slice.608)
  ROOT %dynamic-update-slice.267 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.183, s32[1]{0} %select.332, s32[] %select.664), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.93 (param_0.184: s32[75], param_1.747: s32[], param_2.842: s32[75]) -> s32[75] {
  %param_0.184 = s32[75]{0} parameter(0)
  %constant_4537 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.747 = s32[] parameter(1)
  %constant_5380 = s32[] constant(3)
  %add.1556 = s32[] add(s32[] %param_1.747, s32[] %constant_5380), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1143 = pred[] compare(s32[] %add.1556, s32[] %constant_4537), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5378 = s32[] constant(78)
  %add.1555 = s32[] add(s32[] %param_1.747, s32[] %constant_5378), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.656 = s32[] select(pred[] %compare.1143, s32[] %add.1555, s32[] %add.1556), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.821 = pred[] compare(s32[] %constant_4537, s32[] %select.656), direction=LE
  %constant_4536 = s32[] constant(74)
  %compare.820 = pred[] compare(s32[] %constant_4536, s32[] %select.656), direction=GE
  %and.295 = pred[] and(pred[] %compare.821, pred[] %compare.820)
  %bitcast.151 = pred[1]{0} bitcast(pred[] %and.295)
  %param_2.842 = s32[75]{0} parameter(2)
  %dynamic-slice.717 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.842, s32[] %select.656), dynamic_slice_sizes={1}
  %bitcast.239 = s32[] bitcast(s32[1]{0} %dynamic-slice.717), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1146 = pred[] compare(s32[] %bitcast.239, s32[] %constant_4537), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5387 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1560 = s32[] add(s32[] %bitcast.239, s32[] %constant_5387), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.660 = s32[] select(pred[] %compare.1146, s32[] %add.1560, s32[] %bitcast.239), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.609 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.184, s32[] %select.660), dynamic_slice_sizes={1}
  %dynamic-slice.832 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.184, s32[] %select.656), dynamic_slice_sizes={1}
  %select.334 = s32[1]{0} select(pred[1]{0} %bitcast.151, s32[1]{0} %dynamic-slice.609, s32[1]{0} %dynamic-slice.832)
  ROOT %dynamic-update-slice.268 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.184, s32[1]{0} %select.334, s32[] %select.656), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.96 (param_0.191: s32[75], param_1.745: s32[75], param_2.841: s32[], param_3.892: s32[75]) -> s32[75] {
  %param_0.191 = s32[75]{0} parameter(0)
  %constant_4542 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.745 = s32[75]{0} parameter(1)
  %param_2.841 = s32[] parameter(2)
  %constant_5359 = s32[] constant(2)
  %add.1544 = s32[] add(s32[] %param_2.841, s32[] %constant_5359), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1137 = pred[] compare(s32[] %add.1544, s32[] %constant_4542), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5357 = s32[] constant(77)
  %add.1543 = s32[] add(s32[] %param_2.841, s32[] %constant_5357), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.650 = s32[] select(pred[] %compare.1137, s32[] %add.1543, s32[] %add.1544), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.714 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.745, s32[] %select.650), dynamic_slice_sizes={1}
  %bitcast.237 = s32[] bitcast(s32[1]{0} %dynamic-slice.714), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1136 = pred[] compare(s32[] %bitcast.237, s32[] %constant_4542), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5355 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1542 = s32[] add(s32[] %bitcast.237, s32[] %constant_5355), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.648 = s32[] select(pred[] %compare.1136, s32[] %add.1542, s32[] %bitcast.237), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.826 = pred[] compare(s32[] %constant_4542, s32[] %select.648), direction=LE
  %constant_4541 = s32[] constant(74)
  %compare.825 = pred[] compare(s32[] %constant_4541, s32[] %select.648), direction=GE
  %and.296 = pred[] and(pred[] %compare.826, pred[] %compare.825)
  %bitcast.153 = pred[1]{0} bitcast(pred[] %and.296)
  %param_3.892 = s32[75]{0} parameter(3)
  %dynamic-slice.830 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.892, s32[] %select.650), dynamic_slice_sizes={1}
  %dynamic-slice.611 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.191, s32[] %select.648), dynamic_slice_sizes={1}
  %select.337 = s32[1]{0} select(pred[1]{0} %bitcast.153, s32[1]{0} %dynamic-slice.830, s32[1]{0} %dynamic-slice.611)
  ROOT %dynamic-update-slice.269 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.191, s32[1]{0} %select.337, s32[] %select.648), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.97 (param_0.192: s32[75], param_1.743: s32[], param_2.840: s32[75]) -> s32[75] {
  %param_0.192 = s32[75]{0} parameter(0)
  %constant_4544 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.743 = s32[] parameter(1)
  %constant_5337 = s32[] constant(2)
  %add.1532 = s32[] add(s32[] %param_1.743, s32[] %constant_5337), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1128 = pred[] compare(s32[] %add.1532, s32[] %constant_4544), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5335 = s32[] constant(77)
  %add.1531 = s32[] add(s32[] %param_1.743, s32[] %constant_5335), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.640 = s32[] select(pred[] %compare.1128, s32[] %add.1531, s32[] %add.1532), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.828 = pred[] compare(s32[] %constant_4544, s32[] %select.640), direction=LE
  %constant_4543 = s32[] constant(74)
  %compare.827 = pred[] compare(s32[] %constant_4543, s32[] %select.640), direction=GE
  %and.297 = pred[] and(pred[] %compare.828, pred[] %compare.827)
  %bitcast.154 = pred[1]{0} bitcast(pred[] %and.297)
  %param_2.840 = s32[75]{0} parameter(2)
  %dynamic-slice.712 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.840, s32[] %select.640), dynamic_slice_sizes={1}
  %bitcast.235 = s32[] bitcast(s32[1]{0} %dynamic-slice.712), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1131 = pred[] compare(s32[] %bitcast.235, s32[] %constant_4544), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5344 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1536 = s32[] add(s32[] %bitcast.235, s32[] %constant_5344), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.644 = s32[] select(pred[] %compare.1131, s32[] %add.1536, s32[] %bitcast.235), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.612 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.192, s32[] %select.644), dynamic_slice_sizes={1}
  %dynamic-slice.828 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.192, s32[] %select.640), dynamic_slice_sizes={1}
  %select.338 = s32[1]{0} select(pred[1]{0} %bitcast.154, s32[1]{0} %dynamic-slice.612, s32[1]{0} %dynamic-slice.828)
  ROOT %dynamic-update-slice.270 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.192, s32[1]{0} %select.338, s32[] %select.640), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.100 (param_0.199: s32[75], param_1.741: s32[75], param_2.839: s32[], param_3.887: s32[75]) -> s32[75] {
  %param_0.199 = s32[75]{0} parameter(0)
  %constant_4549 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.741 = s32[75]{0} parameter(1)
  %param_2.839 = s32[] parameter(2)
  %constant_5316 = s32[] constant(1), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %add.1520 = s32[] add(s32[] %param_2.839, s32[] %constant_5316), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1121 = pred[] compare(s32[] %add.1520, s32[] %constant_4549), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5314 = s32[] constant(76)
  %add.1519 = s32[] add(s32[] %param_2.839, s32[] %constant_5314), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.634 = s32[] select(pred[] %compare.1121, s32[] %add.1519, s32[] %add.1520), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.709 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.741, s32[] %select.634), dynamic_slice_sizes={1}
  %bitcast.233 = s32[] bitcast(s32[1]{0} %dynamic-slice.709), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1120 = pred[] compare(s32[] %bitcast.233, s32[] %constant_4549), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5312 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1518 = s32[] add(s32[] %bitcast.233, s32[] %constant_5312), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.633 = s32[] select(pred[] %compare.1120, s32[] %add.1518, s32[] %bitcast.233), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.833 = pred[] compare(s32[] %constant_4549, s32[] %select.633), direction=LE
  %constant_4548 = s32[] constant(74)
  %compare.832 = pred[] compare(s32[] %constant_4548, s32[] %select.633), direction=GE
  %and.298 = pred[] and(pred[] %compare.833, pred[] %compare.832)
  %bitcast.156 = pred[1]{0} bitcast(pred[] %and.298)
  %param_3.887 = s32[75]{0} parameter(3)
  %dynamic-slice.826 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.887, s32[] %select.634), dynamic_slice_sizes={1}
  %dynamic-slice.614 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.199, s32[] %select.633), dynamic_slice_sizes={1}
  %select.342 = s32[1]{0} select(pred[1]{0} %bitcast.156, s32[1]{0} %dynamic-slice.826, s32[1]{0} %dynamic-slice.614)
  ROOT %dynamic-update-slice.271 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.199, s32[1]{0} %select.342, s32[] %select.633), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.101 (param_0.200: s32[75], param_1.739: s32[], param_2.838: s32[75]) -> s32[75] {
  %param_0.200 = s32[75]{0} parameter(0)
  %constant_4551 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.739 = s32[] parameter(1)
  %constant_5294 = s32[] constant(1), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %add.1508 = s32[] add(s32[] %param_1.739, s32[] %constant_5294), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=12}
  %compare.1112 = pred[] compare(s32[] %add.1508, s32[] %constant_4551), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5292 = s32[] constant(76)
  %add.1507 = s32[] add(s32[] %param_1.739, s32[] %constant_5292), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.625 = s32[] select(pred[] %compare.1112, s32[] %add.1507, s32[] %add.1508), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.835 = pred[] compare(s32[] %constant_4551, s32[] %select.625), direction=LE
  %constant_4550 = s32[] constant(74)
  %compare.834 = pred[] compare(s32[] %constant_4550, s32[] %select.625), direction=GE
  %and.299 = pred[] and(pred[] %compare.835, pred[] %compare.834)
  %bitcast.157 = pred[1]{0} bitcast(pred[] %and.299)
  %param_2.838 = s32[75]{0} parameter(2)
  %dynamic-slice.707 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.838, s32[] %select.625), dynamic_slice_sizes={1}
  %bitcast.231 = s32[] bitcast(s32[1]{0} %dynamic-slice.707), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1116 = pred[] compare(s32[] %bitcast.231, s32[] %constant_4551), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %constant_5301 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1512 = s32[] add(s32[] %bitcast.231, s32[] %constant_5301), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.628 = s32[] select(pred[] %compare.1116, s32[] %add.1512, s32[] %bitcast.231), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.615 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.200, s32[] %select.628), dynamic_slice_sizes={1}
  %dynamic-slice.824 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.200, s32[] %select.625), dynamic_slice_sizes={1}
  %select.343 = s32[1]{0} select(pred[1]{0} %bitcast.157, s32[1]{0} %dynamic-slice.615, s32[1]{0} %dynamic-slice.824)
  ROOT %dynamic-update-slice.272 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.200, s32[1]{0} %select.343, s32[] %select.625), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

%fused_computation.104 (param_0.207: s32[75], param_1.737: s32[75], param_2.837: s32[], param_3.882: s32[75]) -> s32[75] {
  %param_0.207 = s32[75]{0} parameter(0)
  %constant_4556 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.737 = s32[75]{0} parameter(1)
  %param_2.837 = s32[] parameter(2)
  %compare.1105 = pred[] compare(s32[] %param_2.837, s32[] %constant_4556), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5270 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1496 = s32[] add(s32[] %param_2.837, s32[] %constant_5270), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.618 = s32[] select(pred[] %compare.1105, s32[] %add.1496, s32[] %param_2.837), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %dynamic-slice.704 = s32[1]{0} dynamic-slice(s32[75]{0} %param_1.737, s32[] %select.618), dynamic_slice_sizes={1}
  %bitcast.229 = s32[] bitcast(s32[1]{0} %dynamic-slice.704), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1104 = pred[] compare(s32[] %bitcast.229, s32[] %constant_4556), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %add.1495 = s32[] add(s32[] %bitcast.229, s32[] %constant_5270), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.617 = s32[] select(pred[] %compare.1104, s32[] %add.1495, s32[] %bitcast.229), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %compare.840 = pred[] compare(s32[] %constant_4556, s32[] %select.617), direction=LE
  %constant_4555 = s32[] constant(74)
  %compare.839 = pred[] compare(s32[] %constant_4555, s32[] %select.617), direction=GE
  %and.300 = pred[] and(pred[] %compare.840, pred[] %compare.839)
  %bitcast.159 = pred[1]{0} bitcast(pred[] %and.300)
  %param_3.882 = s32[75]{0} parameter(3)
  %dynamic-slice.822 = s32[1]{0} dynamic-slice(s32[75]{0} %param_3.882, s32[] %select.618), dynamic_slice_sizes={1}
  %dynamic-slice.617 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.207, s32[] %select.617), dynamic_slice_sizes={1}
  %select.346 = s32[1]{0} select(pred[1]{0} %bitcast.159, s32[1]{0} %dynamic-slice.822, s32[1]{0} %dynamic-slice.617)
  ROOT %dynamic-update-slice.273 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.207, s32[1]{0} %select.346, s32[] %select.617), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
}

%fused_computation.105 (param_0.208: s32[75], param_1.735: s32[], param_2.836: s32[75]) -> s32[75] {
  %param_0.208 = s32[75]{0} parameter(0)
  %constant_4558 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %param_1.735 = s32[] parameter(1)
  %compare.1096 = pred[] compare(s32[] %param_1.735, s32[] %constant_4558), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %constant_5254 = s32[] constant(75), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %add.1488 = s32[] add(s32[] %param_1.735, s32[] %constant_5254), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %select.609 = s32[] select(pred[] %compare.1096, s32[] %add.1488, s32[] %param_1.735), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %compare.842 = pred[] compare(s32[] %constant_4558, s32[] %select.609), direction=LE
  %constant_4557 = s32[] constant(74)
  %compare.841 = pred[] compare(s32[] %constant_4557, s32[] %select.609), direction=GE
  %and.301 = pred[] and(pred[] %compare.842, pred[] %compare.841)
  %bitcast.160 = pred[1]{0} bitcast(pred[] %and.301)
  %param_2.836 = s32[75]{0} parameter(2)
  %dynamic-slice.702 = s32[1]{0} dynamic-slice(s32[75]{0} %param_2.836, s32[] %select.609), dynamic_slice_sizes={1}
  %bitcast.227 = s32[] bitcast(s32[1]{0} %dynamic-slice.702), metadata={op_type="gather" op_name="jit(lu_pivots_to_permutation)/scan/gather[ dimension_numbers=GatherDimensionNumbers(offset_dims=(), collapsed_slice_dims=(0,), start_index_map=(0,))\n                                           slice_sizes=(1,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=6}
  %compare.1100 = pred[] compare(s32[] %bitcast.227, s32[] %constant_4558), direction=LT, metadata={op_type="lt" op_name="jit(lu_pivots_to_permutation)/scan/lt" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %add.1491 = s32[] add(s32[] %bitcast.227, s32[] %constant_5254), metadata={op_type="add" op_name="jit(lu_pivots_to_permutation)/scan/add" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %select.612 = s32[] select(pred[] %compare.1100, s32[] %add.1491, s32[] %bitcast.227), metadata={op_type="select" op_name="jit(lu_pivots_to_permutation)/scan/select" source_file="<ipython-input-5-2a55dee62f1f>" source_line=9}
  %dynamic-slice.618 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.208, s32[] %select.612), dynamic_slice_sizes={1}
  %dynamic-slice.820 = s32[1]{0} dynamic-slice(s32[75]{0} %param_0.208, s32[] %select.609), dynamic_slice_sizes={1}
  %select.347 = s32[1]{0} select(pred[1]{0} %bitcast.160, s32[1]{0} %dynamic-slice.618, s32[1]{0} %dynamic-slice.820)
  ROOT %dynamic-update-slice.274 = s32[75]{0} dynamic-update-slice(s32[75]{0} %param_0.208, s32[1]{0} %select.347, s32[] %select.609), metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
}

ENTRY %jit_lu_pivots_to_permutation.1466 (parameter.6: s32[75]) -> (s32[75]) {
  %constant_11 = s64[] constant(0), metadata={op_type="while" op_name="jit(lu_pivots_to_permutation)/scan/while[ body_nconsts=5\n                                          cond_nconsts=0 ]" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  %copy.64 = s64[] copy(s64[] %constant_11)
  %constant_9 = s32[] constant(0), metadata={op_type="scan" op_name="jit(lu_pivots_to_permutation)/scan[ length=75\n                                    linear=(False, False, False, False, False, False, False, False)\n                                    num_carry=3\n                                    num_consts=5\n                                    reverse=False\n                                    unroll=16 ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=37}
  %copy.63 = s32[] copy(s32[] %constant_9)
  %iota.8 = s32[75]{0} iota(), iota_dimension=0, metadata={op_type="iota" op_name="jit(lu_pivots_to_permutation)/iota[ dimension=0\n                                    dtype=int32\n                                    shape=(75,) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=32}
  %parameter.6 = s32[75]{0} parameter(0)
  %tuple.921 = (s64[], s32[], s32[75]{0}, s32[75]{0}) tuple(s64[] %copy.64, s32[] %copy.63, s32[75]{0} %iota.8, s32[75]{0} %parameter.6)
  %while.135 = (s64[], s32[], s32[75]{0}, s32[75]{0}) while((s64[], s32[], s32[75]{0}, s32[75]{0}) %tuple.921), condition=%cond_computation.860.clone, body=%body_computation.141.clone, backend_config="{\"known_trip_count\":{\"n\":\"4\"}}"
  %get-tuple-element.1914 = s32[75]{0} get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %while.135), index=2, metadata={op_type="while" op_name="jit(lu_pivots_to_permutation)/scan/while[ body_nconsts=5\n                                          cond_nconsts=0 ]" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  %copy.41 = s32[75]{0} copy(s32[75]{0} %get-tuple-element.1914)
  %get-tuple-element.1913 = s32[] get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %while.135), index=1, metadata={op_type="while" op_name="jit(lu_pivots_to_permutation)/scan/while[ body_nconsts=5\n                                          cond_nconsts=0 ]" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  %get-tuple-element.1915 = s32[75]{0} get-tuple-element((s64[], s32[], s32[75]{0}, s32[75]{0}) %while.135), index=3, metadata={op_type="while" op_name="jit(lu_pivots_to_permutation)/scan/while[ body_nconsts=5\n                                          cond_nconsts=0 ]" source_file="<ipython-input-6-f97d46683450>" source_line=1}
  %fusion.105 = s32[75]{0} fusion(s32[75]{0} %copy.41, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.105, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.104 = s32[75]{0} fusion(s32[75]{0} %fusion.105, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1914), kind=kLoop, calls=%fused_computation.104, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.43 = s32[75]{0} copy(s32[75]{0} %fusion.104)
  %fusion.101 = s32[75]{0} fusion(s32[75]{0} %copy.43, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.101, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.100 = s32[75]{0} fusion(s32[75]{0} %fusion.101, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.104), kind=kLoop, calls=%fused_computation.100, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.45 = s32[75]{0} copy(s32[75]{0} %fusion.100)
  %fusion.97 = s32[75]{0} fusion(s32[75]{0} %copy.45, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.97, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.96 = s32[75]{0} fusion(s32[75]{0} %fusion.97, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.100), kind=kLoop, calls=%fused_computation.96, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.47 = s32[75]{0} copy(s32[75]{0} %fusion.96)
  %fusion.93 = s32[75]{0} fusion(s32[75]{0} %copy.47, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.93, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.92 = s32[75]{0} fusion(s32[75]{0} %fusion.93, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.96), kind=kLoop, calls=%fused_computation.92, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.49 = s32[75]{0} copy(s32[75]{0} %fusion.92)
  %fusion.89 = s32[75]{0} fusion(s32[75]{0} %copy.49, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.89, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.88 = s32[75]{0} fusion(s32[75]{0} %fusion.89, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.92), kind=kLoop, calls=%fused_computation.88, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.51 = s32[75]{0} copy(s32[75]{0} %fusion.88)
  %fusion.85 = s32[75]{0} fusion(s32[75]{0} %copy.51, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.85, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.84 = s32[75]{0} fusion(s32[75]{0} %fusion.85, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.88), kind=kLoop, calls=%fused_computation.84, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.53 = s32[75]{0} copy(s32[75]{0} %fusion.84)
  %fusion.81 = s32[75]{0} fusion(s32[75]{0} %copy.53, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.81, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.80 = s32[75]{0} fusion(s32[75]{0} %fusion.81, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.84), kind=kLoop, calls=%fused_computation.80, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.55 = s32[75]{0} copy(s32[75]{0} %fusion.80)
  %fusion.77 = s32[75]{0} fusion(s32[75]{0} %copy.55, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.77, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.76 = s32[75]{0} fusion(s32[75]{0} %fusion.77, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.80), kind=kLoop, calls=%fused_computation.76, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.57 = s32[75]{0} copy(s32[75]{0} %fusion.76)
  %fusion.73 = s32[75]{0} fusion(s32[75]{0} %copy.57, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.73, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.72 = s32[75]{0} fusion(s32[75]{0} %fusion.73, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.76), kind=kLoop, calls=%fused_computation.72, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.59 = s32[75]{0} copy(s32[75]{0} %fusion.72)
  %fusion.69 = s32[75]{0} fusion(s32[75]{0} %copy.59, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.69, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.68 = s32[75]{0} fusion(s32[75]{0} %fusion.69, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.72), kind=kLoop, calls=%fused_computation.68, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  %copy.61 = s32[75]{0} copy(s32[75]{0} %fusion.68)
  %fusion.65 = s32[75]{0} fusion(s32[75]{0} %copy.61, s32[] %get-tuple-element.1913, s32[75]{0} %get-tuple-element.1915), kind=kLoop, calls=%fused_computation.65, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=10}
  %fusion.64 = s32[75]{0} fusion(s32[75]{0} %fusion.65, s32[75]{0} %get-tuple-element.1915, s32[] %get-tuple-element.1913, s32[75]{0} %fusion.68), kind=kLoop, calls=%fused_computation.64, metadata={op_type="scatter" op_name="jit(lu_pivots_to_permutation)/scan/scatter[ dimension_numbers=ScatterDimensionNumbers(update_window_dims=(), inserted_window_dims=(0,), scatter_dims_to_operand_dims=(0,))\n                                            indices_are_sorted=False\n                                            unique_indices=False\n                                            update_consts=(  ) ]" source_file="<ipython-input-5-2a55dee62f1f>" source_line=11}
  ROOT %tuple.1465 = (s32[75]{0}) tuple(s32[75]{0} %fusion.64)
}