Skip to content

Commit e0d755b

Browse files
gchalumpmeta-codesync[bot]
authored andcommitted
fix cpu get_unique_indices (#5151)
Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/2152 Pull Request resolved: #5151 - Use `at::zeros_like()` and `at::zeros()` instead of `at::empty()` for zero-initialized buffers to ensure deterministic cross-platform test comparisons Reviewed By: q10, spcyppt Differential Revision: D87368673 fbshipit-source-id: 73aff4755ed008677364d3ce6c5750c40b0a743c
1 parent ca39ae0 commit e0d755b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fbgemm_gpu/src/split_embeddings_cache/linearize_cache_indices.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ get_unique_indices_cpu_impl(
143143
at::tensor({num_unique}, linear_indices.options().dtype(at::kInt));
144144

145145
// Resize unique_indices to match same size as input
146-
auto unique_indices_output = at::empty_like(linear_indices);
146+
auto unique_indices_output = at::zeros_like(linear_indices);
147147
unique_indices_output.slice(0, 0, num_unique).copy_(unique_indices);
148148

149149
std::optional<Tensor> unique_indices_count = std::nullopt;
@@ -152,7 +152,7 @@ get_unique_indices_cpu_impl(
152152
if (compute_count) {
153153
// Resize counts to match same size as input
154154
unique_indices_count =
155-
at::empty({N}, linear_indices.options().dtype(at::kInt));
155+
at::zeros({N}, linear_indices.options().dtype(at::kInt));
156156
unique_indices_count->slice(0, 0, num_unique).copy_(counts.to(at::kInt));
157157
}
158158

0 commit comments

Comments
 (0)