Skip to content

Commit 693bfaf

Browse files
Style fixes
1 parent a1ae562 commit 693bfaf

5 files changed

Lines changed: 22 additions & 22 deletions

File tree

oneapi-rs-sys/include/usm.hpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@
1414
#include <memory>
1515

1616
namespace sycl_shims::usm {
17-
std::uint8_t* aligned_alloc_device(std::size_t alignment, std::size_t bytes, Queue const &);
18-
std::uint8_t* aligned_alloc_host(std::size_t alignment, std::size_t bytes, Queue const &);
19-
std::uint8_t* aligned_alloc_shared(std::size_t alignment, std::size_t bytes, Queue const &);
17+
std::uint8_t* aligned_alloc_device(std::size_t alignment, std::size_t num_bytes, Queue const &);
18+
std::uint8_t* aligned_alloc_host(std::size_t alignment, std::size_t num_bytes, Queue const &);
19+
std::uint8_t* aligned_alloc_shared(std::size_t alignment, std::size_t num_bytes, Queue const &);
2020
void free(std::uint8_t*, Queue const &);
2121
} // namespace sycl_shims::usm

oneapi-rs-sys/src/usm-sys.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@ pub mod ffi {
1515

1616
extern "C++" {
1717
include!("oneapi-rs-sys/include/usm.hpp");
18-
unsafe fn aligned_alloc_device(alignment: usize, bytes: usize, queue: &Queue) -> Result<*mut u8>;
19-
unsafe fn aligned_alloc_host(alignment: usize, bytes: usize, queue: &Queue) -> Result<*mut u8>;
20-
unsafe fn aligned_alloc_shared(alignment: usize, bytes: usize, queue: &Queue) -> Result<*mut u8>;
18+
unsafe fn aligned_alloc_device(alignment: usize, num_bytes: usize, queue: &Queue) -> Result<*mut u8>;
19+
unsafe fn aligned_alloc_host(alignment: usize, num_bytes: usize, queue: &Queue) -> Result<*mut u8>;
20+
unsafe fn aligned_alloc_shared(alignment: usize, num_bytes: usize, queue: &Queue) -> Result<*mut u8>;
2121
unsafe fn free(ptr: *mut u8, queue: &Queue);
2222
}
2323
}

oneapi-rs-sys/src/usm.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,16 +10,16 @@
1010
#include "oneapi-rs-sys/src/usm-sys.rs.h"
1111

1212
namespace sycl_shims::usm {
13-
std::uint8_t* aligned_alloc_device(std::size_t alignment, std::size_t bytes, Queue const & queue) {
14-
return static_cast<std::uint8_t*>(sycl::aligned_alloc_device(alignment, bytes, queue));
13+
std::uint8_t* aligned_alloc_device(std::size_t alignment, std::size_t num_bytes, Queue const & queue) {
14+
return static_cast<std::uint8_t*>(sycl::aligned_alloc_device(alignment, num_bytes, queue));
1515
}
1616

17-
std::uint8_t* aligned_alloc_host(std::size_t alignment, std::size_t bytes, Queue const & queue) {
18-
return static_cast<std::uint8_t*>(sycl::aligned_alloc_host(alignment, bytes, queue));
17+
std::uint8_t* aligned_alloc_host(std::size_t alignment, std::size_t num_bytes, Queue const & queue) {
18+
return static_cast<std::uint8_t*>(sycl::aligned_alloc_host(alignment, num_bytes, queue));
1919
}
2020

21-
std::uint8_t* aligned_alloc_shared(std::size_t alignment, std::size_t bytes, Queue const & queue) {
22-
return static_cast<std::uint8_t*>(sycl::aligned_alloc_shared(alignment, bytes, queue));
21+
std::uint8_t* aligned_alloc_shared(std::size_t alignment, std::size_t num_bytes, Queue const & queue) {
22+
return static_cast<std::uint8_t*>(sycl::aligned_alloc_shared(alignment, num_bytes, queue));
2323
}
2424

2525
void free(std::uint8_t* memory, Queue const & queue) {

oneapi-rs/src/buffer.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,12 +13,12 @@ use crate::usm::UsmAlloc;
1313
/// The Buffer struct defines a shared array of one, two or three dimensions that can be used
1414
/// by the SYCL kernel. Buffers are templated on the type of their data, and the number of
1515
/// dimensions that the data is stored and accessed through.
16-
16+
///
1717
/// A Buffer does not map to only one underlying backend object, and all SYCL backend memory objects
1818
/// may be temporary for use on a specific device.
19-
19+
///
2020
/// Buffers can be constructed by methods provided by the [`Queue`](`crate::queue::Queue`) class.
21-
21+
///
2222
/// The Buffer struct template takes a template parameter [`UsmAlloc`](`crate::usm::UsmAlloc`) for
2323
/// specifying an allocator which is used by the SYCL runtime when allocating temporary memory on
2424
/// the host.

oneapi-rs/src/usm.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ pub unsafe trait UsmAlloc : Allocator {}
2727
unsafe impl<'a, T: UsmAllocatorKind> UsmAlloc for UsmAllocator<'a, T> {}
2828

2929
pub trait UsmAllocatorKind {
30-
unsafe fn alloc(alignment: usize, bytes: usize, queue: &Queue) -> CxxResult<*mut u8>;
30+
unsafe fn alloc(alignment: usize, num_bytes: usize, queue: &Queue) -> CxxResult<*mut u8>;
3131
}
3232

3333
impl<'a, T: UsmAllocatorKind> From<&'a Queue> for UsmAllocator<'a, T> {
@@ -61,25 +61,25 @@ unsafe impl<T: UsmAllocatorKind> Allocator for UsmAllocator<'_, T> {
6161
pub(crate) struct DeviceAllocator;
6262

6363
impl UsmAllocatorKind for DeviceAllocator {
64-
unsafe fn alloc(alignment: usize, bytes: usize, queue: &Queue) -> CxxResult<*mut u8> {
65-
unsafe { ffi::aligned_alloc_device(alignment, bytes, &queue.0) }
64+
unsafe fn alloc(alignment: usize, num_bytes: usize, queue: &Queue) -> CxxResult<*mut u8> {
65+
unsafe { ffi::aligned_alloc_device(alignment, num_bytes, &queue.0) }
6666
}
6767
}
6868

6969
/// An allocator for Host-side buffers
7070
pub struct HostAllocator;
7171

7272
impl UsmAllocatorKind for HostAllocator {
73-
unsafe fn alloc(alignment: usize, bytes: usize, queue: &Queue) -> CxxResult<*mut u8> {
74-
unsafe { ffi::aligned_alloc_host(alignment, bytes, &queue.0) }
73+
unsafe fn alloc(alignment: usize, num_bytes: usize, queue: &Queue) -> CxxResult<*mut u8> {
74+
unsafe { ffi::aligned_alloc_host(alignment, num_bytes, &queue.0) }
7575
}
7676
}
7777

7878
/// An allocator for shared memory buffers
7979
pub struct SharedAllocator;
8080

8181
impl UsmAllocatorKind for SharedAllocator {
82-
unsafe fn alloc(alignment: usize, bytes: usize, queue: &Queue) -> CxxResult<*mut u8> {
83-
unsafe { ffi::aligned_alloc_shared(alignment, bytes, &queue.0) }
82+
unsafe fn alloc(alignment: usize, num_bytes: usize, queue: &Queue) -> CxxResult<*mut u8> {
83+
unsafe { ffi::aligned_alloc_shared(alignment, num_bytes, &queue.0) }
8484
}
8585
}

0 commit comments

Comments
 (0)