expose build_kernel

LuisaGroup · Oct 3, 2023 · 3550ecd · 3550ecd
1 parent 1c35b28
commit 3550ecd
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/luisa_compute/src/runtime/kernel.rs b/luisa_compute/src/runtime/kernel.rs
@@ -337,7 +337,15 @@ impl KernelBuilder {
             (resource_tracker, cpu_custom_ops, captured)
         })
     }
-    fn build_callable<R: CallableRet>(&mut self, body: impl FnOnce(&mut Self) -> R) -> RawCallable {
+
+
+    /// Don't use this directly
+    /// See [`Callable`] for how to create a callable
+    #[doc(hidden)]
+    pub fn build_callable<R: CallableRet>(
+        &mut self,
+        body: impl FnOnce(&mut Self) -> R,
+    ) -> RawCallable {
         let ret = body(self);
         let ret_type = ret._return();
         let (rt, cpu_custom_ops, captures) = self.collect_module_info();
@@ -381,7 +389,11 @@ impl KernelBuilder {
             }
         })
     }
-    fn build_kernel<S: KernelSignature>(
+
+    /// Don't use this directly
+    /// See [`Kernel`] for how to create a kernel
+    #[doc(hidden)]
+    pub fn build_kernel<S: KernelSignature>(
         &mut self,
         body: impl FnOnce(&mut Self),
     ) -> crate::runtime::KernelDef<S> {