sycl-rs 0.0.9

Rust bindings for SYCL
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

//
// Copyright (C) 2026 Intel Corporation
//
// Under the MIT License or the Apache License v2.0.
// See LICENSE-MIT and LICENSE-APACHE for license information.
// SPDX-License-Identifier: MIT OR Apache-2.0
//

use sycl_rs::prelude::*;

static IOTA_SRC: &str = r#"
#include <sycl/sycl.hpp>
namespace syclext = sycl::ext::oneapi;
namespace syclexp = sycl::ext::oneapi::experimental;

extern "C"
SYCL_EXT_ONEAPI_FUNCTION_PROPERTY((syclexp::nd_range_kernel<1>))
void iota(float start, float *ptr) {
    size_t id = syclext::this_work_item::get_nd_item<1>().get_global_linear_id();
    ptr[id] = start + static_cast<float>(id);
}
"#;

#[tokio::main]
async fn main() -> sycl_rs::Result<()> {
    let mut queue = Queue::new();
    let mut device_array = queue.alloc_device::<f32>(1024)?.await?;

    let kernel = queue
        .get_context()
        .create_kernel_bundle_from_source(IOTA_SRC)?
        .build()?
        .get_kernel("iota")?;

    unsafe {
        queue.launch(
            NdRange::new([1024], [16]),
            &kernel,
            (3.14_f32, &mut device_array),
        )
    }?
    .await?;

    let mut host_array = queue.alloc_host::<f32>(1024)?.await?;

    queue.copy(&device_array, &mut host_array)?.await?;

    for e in host_array.iter() {
        print!("{e} ");
    }
    println!();

    Ok(())
}