Automated Code Change

PiperOrigin-RevId: 714279703
openxla · Jan 11, 2025 · 0db4751 · 0db4751
1 parent 72069fd
commit 0db4751
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/xla/mlir_hlo/mhlo/transforms/unfuse_batch_norm/unfuse_batch_norm.cc b/xla/mlir_hlo/mhlo/transforms/unfuse_batch_norm/unfuse_batch_norm.cc
@@ -362,13 +362,13 @@ class UnfuseBatchNormTrainingPattern
 // In combination with marking such ops as illegal, this allows backends that
 // do not have special support for fused batchnorm to use simpler arithmetic
 // primitives.
-void populateUnfuseBatchNormInferencePattern(MLIRContext *context,
-                                             RewritePatternSet *patterns) {
+static void populateUnfuseBatchNormInferencePattern(
+    MLIRContext *context, RewritePatternSet *patterns) {
   patterns->add<UnfuseBatchNormInferencePattern>(context);
 }
 
-void populateUnfuseBatchNormTrainingPattern(MLIRContext *context,
-                                            RewritePatternSet *patterns) {
+static void populateUnfuseBatchNormTrainingPattern(
+    MLIRContext *context, RewritePatternSet *patterns) {
   patterns->add<UnfuseBatchNormTrainingPattern>(context);
 }
 

diff --git a/xla/mlir_hlo/mhlo/transforms/unfuse_batch_norm/unfuse_batch_norm_pass.cc b/xla/mlir_hlo/mhlo/transforms/unfuse_batch_norm/unfuse_batch_norm_pass.cc
@@ -51,7 +51,7 @@ struct TestUnfuseBatchNormPass
 
 }  // namespace
 
-std::unique_ptr<::mlir::Pass> createTestUnfuseBatchNormPass() {
+static std::unique_ptr<::mlir::Pass> createTestUnfuseBatchNormPass() {
   return std::make_unique<TestUnfuseBatchNormPass>();
 }