use of org.apache.hadoop.fs.LocalFileSystem in project systemml by apache.
the class WriterBinaryBlockParallel method writeBinaryBlockMatrixToHDFS.
@Override
protected void writeBinaryBlockMatrixToHDFS(Path path, JobConf job, FileSystem fs, MatrixBlock src, long rlen, long clen, int brlen, int bclen) throws IOException, DMLRuntimeException {
// estimate output size and number of output blocks (min 1)
int numPartFiles = (int) (OptimizerUtils.estimatePartitionedSizeExactSparsity(rlen, clen, brlen, bclen, src.getNonZeros()) / InfrastructureAnalyzer.getHDFSBlockSize());
numPartFiles = Math.max(numPartFiles, 1);
// determine degree of parallelism
int numThreads = OptimizerUtils.getParallelBinaryWriteParallelism();
numThreads = Math.min(numThreads, numPartFiles);
// fall back to sequential write if dop is 1 (e.g., <128MB) in order to create single file
if (numThreads <= 1) {
super.writeBinaryBlockMatrixToHDFS(path, job, fs, src, rlen, clen, brlen, bclen);
return;
}
// create directory for concurrent tasks
MapReduceTool.createDirIfNotExistOnHDFS(path, DMLConfig.DEFAULT_SHARED_DIR_PERMISSION);
// create and execute write tasks
try {
ExecutorService pool = CommonThreadPool.get(numThreads);
ArrayList<WriteFileTask> tasks = new ArrayList<>();
int blklen = (int) Math.ceil((double) rlen / brlen / numThreads) * brlen;
for (int i = 0; i < numThreads & i * blklen < rlen; i++) {
Path newPath = new Path(path, IOUtilFunctions.getPartFileName(i));
tasks.add(new WriteFileTask(newPath, job, fs, src, i * blklen, Math.min((i + 1) * blklen, rlen), brlen, bclen));
}
// wait until all tasks have been executed
List<Future<Object>> rt = pool.invokeAll(tasks);
pool.shutdown();
// check for exceptions
for (Future<Object> task : rt) task.get();
// delete crc files if written to local file system
if (fs instanceof LocalFileSystem) {
for (int i = 0; i < numThreads & i * blklen < rlen; i++) IOUtilFunctions.deleteCrcFilesFromLocalFileSystem(fs, new Path(path, IOUtilFunctions.getPartFileName(i)));
}
} catch (Exception e) {
throw new IOException("Failed parallel write of binary block input.", e);
}
}
use of org.apache.hadoop.fs.LocalFileSystem in project systemml by apache.
the class WriterMatrixMarketParallel method writeMatrixMarketMatrixToHDFS.
@Override
protected void writeMatrixMarketMatrixToHDFS(Path path, JobConf job, FileSystem fs, MatrixBlock src) throws IOException {
int rlen = src.getNumRows();
// estimate output size and number of output blocks (min 1)
int numPartFiles = (int) (OptimizerUtils.estimateSizeTextOutput(src.getNumRows(), src.getNumColumns(), src.getNonZeros(), OutputInfo.MatrixMarketOutputInfo) / InfrastructureAnalyzer.getHDFSBlockSize());
numPartFiles = Math.max(numPartFiles, 1);
// determine degree of parallelism
int numThreads = OptimizerUtils.getParallelTextWriteParallelism();
numThreads = Math.min(numThreads, numPartFiles);
// fall back to sequential write if dop is 1 (e.g., <128MB) in order to create single file
if (numThreads <= 1) {
super.writeMatrixMarketMatrixToHDFS(path, job, fs, src);
return;
}
// create directory for concurrent tasks
MapReduceTool.createDirIfNotExistOnHDFS(path, DMLConfig.DEFAULT_SHARED_DIR_PERMISSION);
// create and execute tasks
try {
ExecutorService pool = CommonThreadPool.get(numThreads);
ArrayList<WriteMMTask> tasks = new ArrayList<>();
int blklen = (int) Math.ceil((double) rlen / numThreads);
for (int i = 0; i < numThreads & i * blklen < rlen; i++) {
Path newPath = new Path(path, IOUtilFunctions.getPartFileName(i));
tasks.add(new WriteMMTask(newPath, job, fs, src, i * blklen, (int) Math.min((i + 1) * blklen, rlen)));
}
// wait until all tasks have been executed
List<Future<Object>> rt = pool.invokeAll(tasks);
pool.shutdown();
// check for exceptions
for (Future<Object> task : rt) task.get();
// delete crc files if written to local file system
if (fs instanceof LocalFileSystem) {
for (int i = 0; i < numThreads & i * blklen < rlen; i++) IOUtilFunctions.deleteCrcFilesFromLocalFileSystem(fs, new Path(path, IOUtilFunctions.getPartFileName(i)));
}
} catch (Exception e) {
throw new IOException("Failed parallel write of text output.", e);
}
}
use of org.apache.hadoop.fs.LocalFileSystem in project systemml by apache.
the class WriterTextCSVParallel method writeCSVMatrixToHDFS.
@Override
protected void writeCSVMatrixToHDFS(Path path, JobConf job, FileSystem fs, MatrixBlock src, CSVFileFormatProperties csvprops) throws IOException {
// estimate output size and number of output blocks (min 1)
int numPartFiles = (int) (OptimizerUtils.estimateSizeTextOutput(src.getNumRows(), src.getNumColumns(), src.getNonZeros(), OutputInfo.CSVOutputInfo) / InfrastructureAnalyzer.getHDFSBlockSize());
numPartFiles = Math.max(numPartFiles, 1);
// determine degree of parallelism
int numThreads = OptimizerUtils.getParallelTextWriteParallelism();
numThreads = Math.min(numThreads, numPartFiles);
// fall back to sequential write if dop is 1 (e.g., <128MB) in order to create single file
if (numThreads <= 1) {
super.writeCSVMatrixToHDFS(path, job, fs, src, csvprops);
return;
}
// create directory for concurrent tasks
MapReduceTool.createDirIfNotExistOnHDFS(path, DMLConfig.DEFAULT_SHARED_DIR_PERMISSION);
// create and execute tasks
try {
ExecutorService pool = CommonThreadPool.get(numThreads);
ArrayList<WriteCSVTask> tasks = new ArrayList<>();
int rlen = src.getNumRows();
int blklen = (int) Math.ceil((double) rlen / numThreads);
for (int i = 0; i < numThreads & i * blklen < rlen; i++) {
Path newPath = new Path(path, IOUtilFunctions.getPartFileName(i));
tasks.add(new WriteCSVTask(newPath, job, fs, src, i * blklen, (int) Math.min((i + 1) * blklen, rlen), csvprops));
}
// wait until all tasks have been executed
List<Future<Object>> rt = pool.invokeAll(tasks);
pool.shutdown();
// check for exceptions
for (Future<Object> task : rt) task.get();
// delete crc files if written to local file system
if (fs instanceof LocalFileSystem) {
for (int i = 0; i < numThreads & i * blklen < rlen; i++) IOUtilFunctions.deleteCrcFilesFromLocalFileSystem(fs, new Path(path, IOUtilFunctions.getPartFileName(i)));
}
} catch (Exception e) {
throw new IOException("Failed parallel write of csv output.", e);
}
}
use of org.apache.hadoop.fs.LocalFileSystem in project systemml by apache.
the class FrameWriterBinaryBlockParallel method writeBinaryBlockFrameToHDFS.
@Override
protected void writeBinaryBlockFrameToHDFS(Path path, JobConf job, FrameBlock src, long rlen, long clen) throws IOException, DMLRuntimeException {
// estimate output size and number of output blocks (min 1)
int blen = ConfigurationManager.getBlocksize();
int numPartFiles = Math.max((int) (OptimizerUtils.estimatePartitionedSizeExactSparsity(rlen, clen, blen, blen, rlen * clen) / InfrastructureAnalyzer.getHDFSBlockSize()), 1);
// determine degree of parallelism
int numThreads = OptimizerUtils.getParallelBinaryWriteParallelism();
numThreads = Math.min(numThreads, numPartFiles);
// fall back to sequential write if dop is 1 (e.g., <128MB) in order to create single file
if (numThreads <= 1) {
super.writeBinaryBlockFrameToHDFS(path, job, src, rlen, clen);
return;
}
// create directory for concurrent tasks
MapReduceTool.createDirIfNotExistOnHDFS(path, DMLConfig.DEFAULT_SHARED_DIR_PERMISSION);
FileSystem fs = IOUtilFunctions.getFileSystem(path);
// create and execute write tasks
try {
ExecutorService pool = CommonThreadPool.get(numThreads);
ArrayList<WriteFileTask> tasks = new ArrayList<>();
int blklen = (int) Math.ceil((double) rlen / blen / numThreads) * blen;
for (int i = 0; i < numThreads & i * blklen < rlen; i++) {
Path newPath = new Path(path, IOUtilFunctions.getPartFileName(i));
tasks.add(new WriteFileTask(newPath, job, fs, src, i * blklen, Math.min((i + 1) * blklen, (int) rlen), blen));
}
// wait until all tasks have been executed
List<Future<Object>> rt = pool.invokeAll(tasks);
pool.shutdown();
// check for exceptions
for (Future<Object> task : rt) task.get();
// delete crc files if written to local file system
if (fs instanceof LocalFileSystem) {
for (int i = 0; i < numThreads & i * blklen < rlen; i++) IOUtilFunctions.deleteCrcFilesFromLocalFileSystem(fs, new Path(path, IOUtilFunctions.getPartFileName(i)));
}
} catch (Exception e) {
throw new IOException("Failed parallel write of binary block input.", e);
}
}
use of org.apache.hadoop.fs.LocalFileSystem in project systemml by apache.
the class FrameWriterTextCSVParallel method writeCSVFrameToHDFS.
@Override
protected void writeCSVFrameToHDFS(Path path, JobConf job, FrameBlock src, long rlen, long clen, CSVFileFormatProperties csvprops) throws IOException {
// estimate output size and number of output blocks (min 1)
int numPartFiles = Math.max((int) (OptimizerUtils.estimateSizeTextOutput(rlen, clen, rlen * clen, OutputInfo.CSVOutputInfo) / InfrastructureAnalyzer.getHDFSBlockSize()), 1);
// determine degree of parallelism
int numThreads = OptimizerUtils.getParallelTextWriteParallelism();
numThreads = Math.min(numThreads, numPartFiles);
// fall back to sequential write if dop is 1 (e.g., <128MB) in order to create single file
if (numThreads <= 1) {
super.writeCSVFrameToHDFS(path, job, src, rlen, clen, csvprops);
return;
}
// create directory for concurrent tasks
MapReduceTool.createDirIfNotExistOnHDFS(path, DMLConfig.DEFAULT_SHARED_DIR_PERMISSION);
FileSystem fs = IOUtilFunctions.getFileSystem(path, job);
// create and execute tasks
try {
ExecutorService pool = CommonThreadPool.get(numThreads);
ArrayList<WriteFileTask> tasks = new ArrayList<>();
int blklen = (int) Math.ceil((double) rlen / numThreads);
for (int i = 0; i < numThreads & i * blklen < rlen; i++) {
Path newPath = new Path(path, IOUtilFunctions.getPartFileName(i));
tasks.add(new WriteFileTask(newPath, job, fs, src, i * blklen, (int) Math.min((i + 1) * blklen, rlen), csvprops));
}
// wait until all tasks have been executed
List<Future<Object>> rt = pool.invokeAll(tasks);
pool.shutdown();
// check for exceptions
for (Future<Object> task : rt) task.get();
// delete crc files if written to local file system
if (fs instanceof LocalFileSystem) {
for (int i = 0; i < numThreads & i * blklen < rlen; i++) IOUtilFunctions.deleteCrcFilesFromLocalFileSystem(fs, new Path(path, IOUtilFunctions.getPartFileName(i)));
}
} catch (Exception e) {
throw new IOException("Failed parallel write of csv output.", e);
}
}
Aggregations