PTO-BLOCK-MODEL-DISPATCH-CELL-REARRANGEMENT-SCHEMA

1// PTO-UNIT: {"id":"PTO-BLOCK-MODEL-DISPATCH-CELL-REARRANGEMENT-SCHEMA","surface":"block","classification":["model","dispatch","cell-rearrangement-schema"],"depends_on":["PTO-BLOCK-MODEL-DISPATCH-TILE-SCHEMA"]}2
3func ResolveBundleCellRearrangementDestination() => boolean4begin5    var destination_binding: BundleTileBindingIndex = 0;6    var destination_seen = FALSE;7    var source: TileIndex = 0;8    var source_seen = FALSE;9    for binding = 0 to PTO_BUNDLE_TILE_BINDING_COUNT - 1 looplimit 16 do10        if _BundleTileBindings[[binding]].valid then11            if !destination_seen &&12               _BundleTileBindings[[binding]].destination_valid then13                destination_binding = binding as BundleTileBindingIndex;14                destination_seen = TRUE;15            end;16            if !source_seen &&17               _BundleTileBindings[[binding]].source0_valid then18                source = BundleTileSourceIndex(19                    binding as BundleTileBindingIndex, FALSE);20                source_seen = TRUE;21            elsif !source_seen &&22                  _BundleTileBindings[[binding]].source1_valid then23                source = BundleTileSourceIndex(24                    binding as BundleTileBindingIndex, TRUE);25                source_seen = TRUE;26            end;27        end;28    end;29    if !destination_seen || !source_seen ||30       !TileCubeDescriptorLegal(_Tiles[[source]]) then31        SetFault(Fault_TileLegality, ReadTPC());32        return FALSE;33    end;34    let binding = _BundleTileBindings[[destination_binding]];35    let source_tile = _Tiles[[source]];36    let capacity_bytes = BundleLocalDestinationAllocationBytes(37        destination_binding);38    if !TileCubeDescriptorShapeLegal(capacity_bytes,39           source_tile.valid_rows, source_tile.valid_columns,40           source_tile.data_type, source_tile.layout) then41        SetFault(Fault_TileAllocation, ReadTPC());42        return FALSE;43    end;44    let hand = UInt(binding.destination_hand);45    var resolved: TileIndex = 0;46    var found = FALSE;47    for offset = 0 to 15 looplimit 16 do48        let raw_index = hand * 16 + offset;49        if !found && !_Tiles[[raw_index]].allocated then50            resolved = raw_index as TileIndex;51            found = TRUE;52        end;53    end;54    if !found || !LocalTileAllocationFitsExcept(55           resolved, binding.pe_mask, capacity_bytes) then56        SetFault(Fault_TileAllocation, ReadTPC());57        return FALSE;58    end;59    let configured = ConfigureCubeTileForMask(60        resolved, capacity_bytes, source_tile.valid_rows,61        source_tile.valid_columns, source_tile.data_type,62        source_tile.layout, TileLocation_Matrix, binding.pe_mask);63    if !configured then64        SetFault(Fault_TileAllocation, ReadTPC());65        return FALSE;66    end;67    _BundleTileBindings[[destination_binding]].destination = resolved;68    _BundleTileBindings[[destination_binding]].destination_allocated_by_bundle =69        TRUE;70    return TRUE;71end;72
73pure func TileOperationUsesCellRearrangementSchema(74    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean75begin76    let decoded = TileOperationOfIndex(operation);77    return decoded == TileOperation_TPERMUTE ||78           decoded == TileOperation_TSHUF ||79           decoded == TileOperation_TPACK ||80           decoded == TileOperation_TUNPACK;81end;82
83readonly func SelectedBundleCellRearrangementSchemaLegal(84    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean85begin86    if !TileOperationUsesCellRearrangementSchema(operation) then return TRUE; end;87    if SelectedBundleTileMaskIsZero() then return TRUE; end;88    let decoded = TileOperationOfIndex(operation);89    if decoded == TileOperation_TPERMUTE then90        if BundleTileBindingCount() != 2 || BundleSharedBindingCount() != 0 ||91           _BundleScalarBindings[[0]].valid then return FALSE; end;92        let first = _BundleTileBindings[[0]];93        let second = _BundleTileBindings[[1]];94        return !first.destination_valid && first.source0_valid &&95               first.source1_valid && !first.last && second.destination_valid &&96               !second.destination_allocated_by_bundle && second.source0_valid &&97               !second.source1_valid && second.last &&98               second.source0 != first.source0 &&99               second.source0 != first.source1 &&100               BundleTileDestinationSizeLegal(1);101    end;102    if BundleTileBindingCount() != 1 || BundleSharedBindingCount() != 0 ||103       !_BundleScalarBindings[[0]].valid ||104       _BundleScalarBindings[[0]].source1 != 0 ||105       _BundleScalarBindings[[0]].source2 != 0 ||106       _BundleScalarBindings[[0]].destination != 0 then107        return FALSE;108    end;109    let binding = _BundleTileBindings[[0]];110    if !binding.destination_valid || binding.destination_allocated_by_bundle ||111       !BundleTileDestinationSizeLegal(0) || !binding.last then112        return FALSE;113    end;114    if decoded == TileOperation_TUNPACK then115        return binding.source0_valid && !binding.source1_valid;116    end;117    return binding.source0_valid && binding.source1_valid;118end;119