Skip to content

Commit 17536db

Browse files
committed
Fix
1 parent 2e7f266 commit 17536db

2 files changed

Lines changed: 17 additions & 3 deletions

File tree

cpp/oneapi/dal/algo/connected_components/detail/vertex_partitioning_default_kernel.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,4 +36,14 @@ template struct afforest<float,
3636
task::vertex_partitioning,
3737
dal::preview::detail::topology<std::int32_t>>;
3838

39+
vertex_partitioning_result<task::vertex_partitioning> run_afforest(
40+
const dal::detail::host_policy& ctx,
41+
const detail::descriptor_base<task::vertex_partitioning>& desc,
42+
const dal::preview::detail::topology<std::int32_t>& t,
43+
byte_alloc_iface* alloc) {
44+
return afforest<float,
45+
task::vertex_partitioning,
46+
dal::preview::detail::topology<std::int32_t>>{}(ctx, desc, t, alloc);
47+
}
48+
3949
} // namespace oneapi::dal::preview::connected_components::detail

cpp/oneapi/dal/algo/connected_components/detail/vertex_partitioning_default_kernel.hpp

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,12 @@ struct vertex_partitioning_kernel_cpu<method::afforest,
7878
}
7979
};
8080

81+
ONEDAL_EXPORT vertex_partitioning_result<task::vertex_partitioning> run_afforest(
82+
const dal::detail::host_policy& ctx,
83+
const detail::descriptor_base<task::vertex_partitioning>& desc,
84+
const dal::preview::detail::topology<std::int32_t>& t,
85+
byte_alloc_iface* alloc);
86+
8187
template <typename Allocator>
8288
struct vertex_partitioning_kernel_cpu<method::afforest,
8389
task::vertex_partitioning,
@@ -93,9 +99,7 @@ struct vertex_partitioning_kernel_cpu<method::afforest,
9399
if (vertex_count == 0) {
94100
return vertex_partitioning_result<task::vertex_partitioning>();
95101
}
96-
return afforest<float,
97-
task::vertex_partitioning,
98-
dal::preview::detail::topology<std::int32_t>>{}(ctx, desc, t, &alloc_con);
102+
return run_afforest(ctx, desc, t, &alloc_con);
99103
}
100104
};
101105

0 commit comments

Comments
 (0)