Skip to content

Commit 4709555

Browse files
chsiggcopybara-github
authored andcommitted
Fix command line arguments in lhlo-tfrt-opt.
[NCF] Upper-case `tfrt::gpu::PopulateGpuToTfrtGpuPasses/RegisterPasses()`. PiperOrigin-RevId: 444929936
1 parent 78537f1 commit 4709555

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

backends/gpu/include/tfrt/gpu/passes/passes.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,10 +105,10 @@ void PopulateMemrefConversionPatterns(mlir::RewritePatternSet& patterns,
105105

106106
// Adds passes to convert from MLIR's gpu and async dialects to TFRT. Adds
107107
// !tfrt.chain result and !tfrt.chain, !tfrt_gpu.stream arguments to functions.
108-
void populateGpuToTfrtGpuPasses(mlir::OpPassManager& pm);
108+
void PopulateGpuToTfrtGpuPasses(mlir::OpPassManager& pm);
109109

110110
// Registers all tfrt gpu passes.
111-
void registerPasses();
111+
void RegisterPasses();
112112

113113
// Creates a pass which adds a function returning the entry point information
114114
// for the gpu executor.

backends/gpu/lib/passes/gpu_to_tfrt_passes.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1726,7 +1726,7 @@ TypeConverter CreateMemrefToTfrtGpuConverter() {
17261726
return converter;
17271727
}
17281728

1729-
void populateGpuToTfrtGpuPasses(OpPassManager &pm) {
1729+
void PopulateGpuToTfrtGpuPasses(OpPassManager &pm) {
17301730
pm.addPass(std::make_unique<AddChainAndStreamToFuncPass>());
17311731
pm.addPass(std::make_unique<ConvertAsyncToChainAndEventPass>());
17321732
pm.addPass(std::make_unique<ConvertGpuToTfrtGpuPass>());
@@ -1735,7 +1735,7 @@ void populateGpuToTfrtGpuPasses(OpPassManager &pm) {
17351735
pm.addPass(std::make_unique<HoistingPass>());
17361736
}
17371737

1738-
void registerPasses() {
1738+
void RegisterPasses() {
17391739
registerPass([] { return CreateStreamifyOpsPass({}); });
17401740
PassRegistration<AddChainAndStreamToFuncPass>();
17411741
PassRegistration<ConvertAsyncToChainAndEventPass>();
@@ -1748,7 +1748,7 @@ void registerPasses() {
17481748
PassPipelineRegistration<>(
17491749
"gpu-to-tfrt-gpu",
17501750
"Pass pipeline to convert from MLIR's gpu and async dialects to TFRT.",
1751-
[](OpPassManager &pm) { tfrt::gpu::populateGpuToTfrtGpuPasses(pm); });
1751+
[](OpPassManager &pm) { tfrt::gpu::PopulateGpuToTfrtGpuPasses(pm); });
17521752
}
17531753

17541754
} // namespace gpu

backends/gpu/tools/tfrt_gpu_opt/tfrt_gpu_opt.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ int main(int argc, char **argv) {
5050
mlir::gpu::GPUDialect, mlir::memref::MemRefDialect,
5151
tfrt::compiler::TFRTDialect, tfrt::gpu::GpuDialect,
5252
tfrt::test::TestDialect>();
53-
tfrt::gpu::registerPasses();
53+
tfrt::gpu::RegisterPasses();
5454

5555
return mlir::asMainReturnCode(
5656
mlir::MlirOptMain(argc, argv, "TFRT pass driver\n", registry));

0 commit comments

Comments
 (0)