This is an archive of the discontinued LLVM Phabricator instance.

Differential D117756

Print the `// ----` separator between modules when using -split-input-file with mlir-opt
ClosedPublic

Authored by mehdi_amini on Jan 20 2022, 1:48 AM.

Details

Reviewers
rriddle
clementval
Commits
rG26167cae4515: Print the `// ----` separator between modules when using -split-input-file with…
Summary

This allows to pipe sequences of mlir-opt -split-input-file | mlir-opt -split-input-file.

Depends On D117750

Diff Detail

Revision Contents

PathSize
mlir/
lib/
Support/
10 lines
test/
Dialect/
OpenACC/
4 lines

Diff 401854

mlir/lib/Support/MlirOptMain.cpp

Show First 20 Lines • Show All 152 Lines • ▼ Show 20 LinesLogicalResult mlir::MlirOptMain(raw_ostream &outputStream,
MLIRContext threadPoolCtx; MLIRContext threadPoolCtx;
if (threadPoolCtx.isMultithreadingEnabled()) if (threadPoolCtx.isMultithreadingEnabled())
threadPool = &threadPoolCtx.getThreadPool(); threadPool = &threadPoolCtx.getThreadPool();
if (splitInputFile) if (splitInputFile)
return splitAndProcessBuffer( return splitAndProcessBuffer(
std::move(buffer), std::move(buffer),
[&](std::unique_ptr<MemoryBuffer> chunkBuffer, raw_ostream &os) { [&](std::unique_ptr<MemoryBuffer> chunkBuffer, raw_ostream &os) {
return processBuffer(os, std::move(chunkBuffer), verifyDiagnostics, LogicalResult result = processBuffer(
verifyPasses, allowUnregisteredDialects, os, std::move(chunkBuffer), verifyDiagnostics, verifyPasses,
preloadDialectsInContext, passManagerSetupFn, allowUnregisteredDialects, preloadDialectsInContext,
registry, threadPool); passManagerSetupFn, registry, threadPool);
os << "// -----\n";
return result;
}, },
outputStream); outputStream);
return processBuffer(outputStream, std::move(buffer), verifyDiagnostics, return processBuffer(outputStream, std::move(buffer), verifyDiagnostics,
verifyPasses, allowUnregisteredDialects, verifyPasses, allowUnregisteredDialects,
preloadDialectsInContext, passManagerSetupFn, registry, preloadDialectsInContext, passManagerSetupFn, registry,
threadPool); threadPool);
} }
▲ Show 20 Lines • Show All 128 Lines • Show Last 20 Lines

mlir/test/Dialect/OpenACC/ops.mlir

// RUN: mlir-opt -split-input-file %s | FileCheck %s // RUN: mlir-opt -split-input-file %s | FileCheck %s
// Verify the printed output can be parsed. // Verify the printed output can be parsed.
// RUN: mlir-opt -split-input-file %s | mlir-opt -allow-unregistered-dialect | FileCheck %s // RUN: mlir-opt -split-input-file %s | mlir-opt -split-input-file | FileCheck %s
// Verify the generic form can be parsed. // Verify the generic form can be parsed.
// RUN: mlir-opt -split-input-file -mlir-print-op-generic %s | mlir-opt -allow-unregistered-dialect | FileCheck %s // RUN: mlir-opt -split-input-file -mlir-print-op-generic %s | mlir-opt -split-input-file | FileCheck %s
func @compute1(%A: memref<10x10xf32>, %B: memref<10x10xf32>, %C: memref<10x10xf32>) -> memref<10x10xf32> { func @compute1(%A: memref<10x10xf32>, %B: memref<10x10xf32>, %C: memref<10x10xf32>) -> memref<10x10xf32> {
%c0 = arith.constant 0 : index %c0 = arith.constant 0 : index
%c10 = arith.constant 10 : index %c10 = arith.constant 10 : index
%c1 = arith.constant 1 : index %c1 = arith.constant 1 : index
%async = arith.constant 1 : i64 %async = arith.constant 1 : i64
acc.parallel async(%async: i64) { acc.parallel async(%async: i64) {
▲ Show 20 Lines • Show All 713 Lines • Show Last 20 Lines