refactor(cndrv): Blob 移动到 memory，缩小范围

Signed-off-by: YdrMaster <ydrml@hotmail.com>
InfiniTensor · Jul 26, 2024 · 8278816 · 8278816
1 parent fc35abc
commit 8278816
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/cndrv/src/lib.rs b/cndrv/src/lib.rs
@@ -82,10 +82,7 @@ pub use notifier::{Notifier, NotifierSpore};
 pub use queue::{Queue, QueueSpore};
 pub use version::{driver_version, library_version, Version};
 
-struct Blob<P> {
-    ptr: P,
-    len: usize,
-}
+
 
 #[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)]
 #[repr(transparent)]

diff --git a/cndrv/src/memory.rs b/cndrv/src/memory.rs
@@ -1,4 +1,4 @@
-use crate::{bindings::CNaddr, Blob, CurrentCtx, Queue};
+use crate::{bindings::CNaddr, CurrentCtx, Queue};
 use context_spore::{impl_spore, AsRaw};
 use std::{
     alloc::Layout,
@@ -63,6 +63,11 @@ impl Queue<'_> {
     }
 }
 
+struct Blob<P> {
+    ptr: P,
+    len: usize,
+}
+
 impl_spore!(DevMem and DevMemSpore by (CurrentCtx, Blob<CNaddr>));
 
 impl CurrentCtx {