Propagate visibility to CubeLaunch-generated structures. (#70)

tracel-ai · Aug 23, 2024 · 4572305 · 4572305
1 parent e158a93
commit 4572305
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/cubecl-macros/src/codegen_type/base.rs b/crates/cubecl-macros/src/codegen_type/base.rs
@@ -54,9 +54,10 @@ impl TypeCodegen {
         }
 
         let generics = self.generics.all_definitions();
+        let vis = &self.vis;
 
         quote! {
-            struct #name #generics {
+            #vis struct #name #generics {
                 #fields
             }
         }