PTO-BLOCK-MODEL-DISPATCH-REDUCTION-SCHEMA

1// PTO-UNIT: {"id":"PTO-BLOCK-MODEL-DISPATCH-REDUCTION-SCHEMA","surface":"block","classification":["model","dispatch","reduction-schema"],"depends_on":["PTO-BLOCK-MODEL-DISPATCH-COMPARISON-SCHEMA"]}2
3pure func TileOperationUsesClosedRowReductionSchema(4    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean5begin6    let decoded = TileOperationOfIndex(operation);7    return decoded == TileOperation_TROWSUM ||8           decoded == TileOperation_TROWPROD ||9           decoded == TileOperation_TROWMIN ||10           decoded == TileOperation_TROWMAX ||11           decoded == TileOperation_TROWARGMIN ||12           decoded == TileOperation_TROWARGMAX;13end;14
15pure func TileOperationUsesClosedColumnReductionSchema(16    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean17begin18    let decoded = TileOperationOfIndex(operation);19    return decoded == TileOperation_TCOLSUM ||20           decoded == TileOperation_TCOLPROD ||21           decoded == TileOperation_TCOLMIN ||22           decoded == TileOperation_TCOLMAX ||23           decoded == TileOperation_TCOLARGMIN ||24           decoded == TileOperation_TCOLARGMAX;25end;26
27pure func TileOperationUsesClosedReductionSchema(28    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean29begin30    return TileOperationUsesClosedRowReductionSchema(operation) ||31           TileOperationUsesClosedColumnReductionSchema(operation);32end;33
34pure func TileReductionOperationReturnsIndex(35    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean36begin37    let decoded = TileOperationOfIndex(operation);38    return decoded == TileOperation_TROWARGMIN ||39           decoded == TileOperation_TROWARGMAX ||40           decoded == TileOperation_TCOLARGMIN ||41           decoded == TileOperation_TCOLARGMAX;42end;43
44readonly func SelectedBundleClosedReductionSchemaLegal(45    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean46begin47    if !TileOperationUsesClosedReductionSchema(operation) then48        return TRUE;49    end;50    if BundleTileBindingCount() != 1 ||51       BundleSharedBindingCount() != 0 ||52       _BundleScalarBindings[[0]].valid ||53       !SelectedBundleComparisonDimensionsLegal() then54        return FALSE;55    end;56
57    let binding = _BundleTileBindings[[0]];58    if !binding.destination_valid ||59       binding.destination_allocated_by_bundle ||60       !BundleTileDestinationSizeLegal(0) ||61       !binding.source0_valid ||62       binding.source1_valid ||63       !binding.last then64        return FALSE;65    end;66
67    let source = BundleTileSourceIndex(0, FALSE);68    let data_type = TileDataTypeFromEncoding(69        CurrentBundleTileOperationDataTypeCode()70            as TileDataTypeEncoding);71    return TileVecArithmeticDataTypeSupported(data_type) &&72           TileDescriptorLegal(source) &&73           _Tiles[[source]].storage_kind == TileStorage_Numeric &&74           _Tiles[[source]].data_type == data_type &&75           _Tiles[[source]].layout == TileLayout_RowMajor &&76           _Tiles[[source]].valid_rows > 0 &&77           _Tiles[[source]].valid_columns > 0 &&78           SelectedBundleComparisonShapeMatches(source) &&79           TileSourceContentsDefined(source) &&80           TileSourceEncodingsValid(source);81end;82