| 77 | } |
| 78 | |
| 79 | void launch_param_update_half(const float* input, sycl::half* output, int size, sycl::queue* stream) |
| 80 | { |
| 81 | int threads = 1024; |
| 82 | size /= 2; |
| 83 | sycl::range<3> grid_dim(1, 1, (size - 1) / threads + 1); |
| 84 | sycl::range<3> block_dim(1, 1, threads); |
| 85 | |
| 86 | { |
| 87 | has_capability_or_fail(stream->get_device(), {sycl::aspect::fp16}); |
| 88 | stream->parallel_for( |
| 89 | sycl::nd_range<3>(grid_dim * block_dim, block_dim), |
| 90 | [=](sycl::nd_item<3> item_ct1) { param_update_kernel_half(input, output, size); }); |
| 91 | } |
| 92 | } |
nothing calls this directly
no test coverage detected