Examples with CompIntFloatRowUpdateSplit com.tencent.angel.psagent.matrix.oplog.cache.CompIntFloatRowUpdateSplit used on opensource projects

Example 1 with CompIntFloatRowUpdateSplit

use of com.tencent.angel.psagent.matrix.oplog.cache.CompIntFloatRowUpdateSplit in project angel by Tencent.

the class CompIntFloatVectorSplitter method split.

@Override
public Map<PartitionKey, RowUpdateSplit> split(Vector vector, List<PartitionKey> parts) {
    IntFloatVector[] vecParts = ((CompIntFloatVector) vector).getPartitions();
    assert vecParts.length == parts.size();
    Map<PartitionKey, RowUpdateSplit> updateSplitMap = new HashMap<>(parts.size());
    for (int i = 0; i < vecParts.length; i++) {
        updateSplitMap.put(parts.get(i), new CompIntFloatRowUpdateSplit(vector.getRowId(), vecParts[i], (int) (parts.get(i).getEndCol() - parts.get(i).getStartCol())));
    }
    return updateSplitMap;
}