File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed
dpctl/tensor/libtensor/include/kernels/sorting Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change 34
34
#include < utility>
35
35
#include < vector>
36
36
37
- #include " kernels/dpctl_tensor_types.hpp"
38
37
#include < sycl/sycl.hpp>
39
38
39
+ #include " kernels/dpctl_tensor_types.hpp"
40
+ #include " utils/sycl_alloc_utils.hpp"
41
+
40
42
namespace dpctl
41
43
{
42
44
namespace tensor
@@ -1519,8 +1521,10 @@ sycl::event parallel_radix_sort_impl(sycl::queue &exec_q,
1519
1521
sort_ev = exec_q.submit ([=](sycl::handler &cgh) {
1520
1522
cgh.depends_on (sort_ev);
1521
1523
const sycl::context &ctx = exec_q.get_context ();
1524
+
1525
+ using dpctl::tensor::alloc_utils::sycl_free_noexcept;
1522
1526
cgh.host_task (
1523
- [ctx, count_ptr]() { sycl::free (count_ptr, ctx); });
1527
+ [ctx, count_ptr]() { sycl_free_noexcept (count_ptr, ctx); });
1524
1528
});
1525
1529
1526
1530
return sort_ev;
@@ -1572,9 +1576,11 @@ sycl::event parallel_radix_sort_impl(sycl::queue &exec_q,
1572
1576
cgh.depends_on (sort_ev);
1573
1577
1574
1578
const sycl::context &ctx = exec_q.get_context ();
1579
+
1580
+ using dpctl::tensor::alloc_utils::sycl_free_noexcept;
1575
1581
cgh.host_task ([ctx, count_ptr, tmp_arr]() {
1576
- sycl::free (tmp_arr, ctx);
1577
- sycl::free (count_ptr, ctx);
1582
+ sycl_free_noexcept (tmp_arr, ctx);
1583
+ sycl_free_noexcept (count_ptr, ctx);
1578
1584
});
1579
1585
});
1580
1586
}
You can’t perform that action at this time.
0 commit comments