Add kernel argument trait

Author: szymon-zadworny

oneapi-rs/src/buffer.rs

@@ -15,11 +15,11 @@ use std::{
     task::{Context, Poll},
 };
 
+use bytemuck::Pod;
 use pin_project::pin_project;
 
 use crate::{
-    event::{Event, EventFuture},
-    usm::UsmAlloc,
+    event::{Event, EventFuture}, kernel_bundle::KernelArgument, usm::UsmAlloc,
 };
 
 /// The Buffer struct defines a shared array of one, two or three dimensions that can be used
@@ -138,3 +138,14 @@ impl<T, A: UsmAlloc> IntoFuture for EnqueuedBuffer<T, A> {
         }
     }
 }
+
+unsafe impl<T: Pod, A: UsmAlloc> KernelArgument for Buffer<T, A> {
+    unsafe fn as_raw_arg(&self) -> &[u8] {
+        unsafe {
+            slice::from_raw_parts(
+                self.data.as_ptr() as *mut u8, 
+                std::mem::size_of::<*mut u8>()
+            )
+        }
+    }
+}

oneapi-rs/src/kernel_bundle.rs

@@ -6,6 +6,7 @@
 // SPDX-License-Identifier: MIT OR Apache-2.0
 //
 
+use bytemuck::Pod;
 use oneapi_rs_sys::{kernel_bundle::ffi, types};
 
 pub struct SourceKernelBundle(pub(crate) cxx::UniquePtr<types::ffi::SourceKernelBundle>);
@@ -29,3 +30,13 @@ impl From<cxx::UniquePtr<types::ffi::ExecutableKernelBundle>> for ExecutableKern
         Self(value)
     }
 }
+
+pub unsafe trait KernelArgument {
+    unsafe fn as_raw_arg(&self) -> &[u8];
+}
+
+unsafe impl<T: Pod> KernelArgument for T {
+    unsafe fn as_raw_arg(&self) -> &[u8] {
+        bytemuck::bytes_of(self)
+    }
+}