Remove unsafe in example, remove todo from bc

elftausend · Nov 17, 2024 · 2712285 · 2712285
1 parent 2bb7672
commit 2712285
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/examples/custom_module.rs b/examples/custom_module.rs
@@ -76,7 +76,7 @@ where
     S: Shape,
 {
     #[inline]
-    unsafe fn retrieve_entry<const NUM_PARENTS: usize>(
+    fn retrieve_entry<const NUM_PARENTS: usize>(
         &'a self,
         device: &D,
         len: usize,
@@ -104,7 +104,7 @@ where
         self.mods.on_retrieve_finish(len, parents, retrieved_buf)
     }
 
-    unsafe fn retrieve<const NUM_PARENTS: usize>(
+    fn retrieve<const NUM_PARENTS: usize>(
         &self,
         device: &D,
         len: usize,

diff --git a/src/cache/borrow_cache.rs b/src/cache/borrow_cache.rs
@@ -117,8 +117,7 @@ impl BorrowCache {
         D: Alloc<T> + 'static,
         S: Shape,
     {
-        todo!()
-        // self.cache.get(&id)?.downcast_ref()
+        self.get_buf(_dev, id).ok()
     }
 
     #[inline]