File tree Expand file tree Collapse file tree 1 file changed +21
-0
lines changed Expand file tree Collapse file tree 1 file changed +21
-0
lines changed Original file line number Diff line number Diff line change @@ -30,6 +30,27 @@ def get_spirv_abspath(fn):
30
30
return spirv_file
31
31
32
32
33
+ # The kernel in the SPIR-V file used in this test was generated from the
34
+ # following SYCL source code:
35
+ # #include <sycl/sycl.hpp>
36
+ # using namespace sycl;
37
+ # namespace syclexp = sycl::ext::oneapi::experimental;
38
+ # namespace syclext = sycl::ext::oneapi;
39
+ # using data_t = int32_t;
40
+ #
41
+ # extern "C" SYCL_EXTERNAL
42
+ # SYCL_EXT_ONEAPI_FUNCTION_PROPERTY((syclexp::nd_range_kernel<1>))
43
+ # void local_mem_kernel(data_t* in, data_t* out,
44
+ # syclexp::work_group_memory<data_t> mem){
45
+ # auto* local_mem = &mem;
46
+ # auto item = syclext::this_work_item::get_nd_item<1>();
47
+ # size_t global_id = item.get_global_linear_id();
48
+ # size_t local_id = item.get_local_linear_id();
49
+ # local_mem[local_id] = in[global_id];
50
+ # out[global_id] = local_mem[local_id];
51
+ # }
52
+
53
+
33
54
def test_submit_work_group_memory ():
34
55
if not dpctl .experimental .WorkGroupMemory .is_available ():
35
56
pytest .skip ("Work group memory extension not supported" )
You can’t perform that action at this time.
0 commit comments