Declare set_len unsafe for issue #69

kenba · Dec 15, 2024 · 5c8e67b · 5c8e67b
1 parent 96df90b
commit 5c8e67b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/svm.rs b/src/svm.rs
@@ -315,7 +315,7 @@ impl<'a, T> SvmVec<'a, T> {
     ///
     /// # Safety
     /// May fail to grow buf if memory is not available for new_len.
-    pub fn set_len(&mut self, new_len: usize) -> Result<()> {
+    pub unsafe fn set_len(&mut self, new_len: usize) -> Result<()> {
         if self.cap() < new_len {
             self.buf.grow(new_len)?;
         }

diff --git a/tests/integration_test.rs b/tests/integration_test.rs
@@ -281,11 +281,11 @@ fn test_opencl_svm_example() -> Result<()> {
             println!("kernel execution duration (ns): {}", duration);
         } else {
             // !is_fine_grained_svm
-            // unsafe { ones.set_len(ARRAY_SIZE) };
-            // unsafe { sums.set_len(ARRAY_SIZE) };
 
-            // Map the input SVM vectors, before setting their data
+            // Resize and map the input SVM vectors, before setting their data
             unsafe {
+                ones.set_len(ARRAY_SIZE)?;
+                sums.set_len(ARRAY_SIZE)? ;
                 queue.enqueue_svm_map(CL_BLOCKING, CL_MAP_WRITE, &mut ones, &[])?;
                 queue.enqueue_svm_map(CL_BLOCKING, CL_MAP_WRITE, &mut sums, &[])?;
             }