From cdfe1bd4910b73b50e6751c3ff19d1f3aa170a6a Mon Sep 17 00:00:00 2001
From: Brian Jesse <brian.jesse@gmail.com>
Date: Fri, 7 Feb 2025 01:19:50 +0000
Subject: [PATCH] Remove pub from wgsl functions

---
 .../transformer/to_wgsl_syntax/remove_pub_from_struct_def.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/bevy_gpu_compute_macro/src/transformer/to_wgsl_syntax/remove_pub_from_struct_def.rs b/bevy_gpu_compute_macro/src/transformer/to_wgsl_syntax/remove_pub_from_struct_def.rs
index f1292d7..f9da5e1 100644
--- a/bevy_gpu_compute_macro/src/transformer/to_wgsl_syntax/remove_pub_from_struct_def.rs
+++ b/bevy_gpu_compute_macro/src/transformer/to_wgsl_syntax/remove_pub_from_struct_def.rs
@@ -3,6 +3,11 @@ use syn::{Field, Visibility, visit_mut::VisitMut};
 pub struct PubRemover {}
 
 impl VisitMut for PubRemover {
+    fn visit_item_fn_mut(&mut self, i: &mut syn::ItemFn) {
+        syn::visit_mut::visit_item_fn_mut(self, i);
+        i.vis = Visibility::Inherited;
+    }
+
     fn visit_field_mut(&mut self, i: &mut Field) {
         syn::visit_mut::visit_field_mut(self, i);
         i.vis = Visibility::Inherited;