Skip to content

Commit 5a26d1d

Browse files
Remove unnecessary sentinel node in coo tensors while filling row pointers.
PiperOrigin-RevId: 762458372
1 parent d7d0f12 commit 5a26d1d

File tree

2 files changed

+146
-149
lines changed

2 files changed

+146
-149
lines changed

jax_tpu_embedding/sparsecore/lib/core/input_preprocessing.cc

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -329,9 +329,6 @@ void PreprocessInputForStackedTablePerLocalDevice(
329329
stacked_table_name, allow_id_dropping, num_sc_per_device,
330330
total_num_coo_tensors, total_max_ids_per_sc,
331331
total_max_unique_ids_per_sc, required_buffer_size_per_sc);
332-
for (int i = 0; i < num_sc_per_device; ++i) {
333-
coo_tensors_by_id[i].emplace_back(batch_size_per_sc * (i + 1), 0, 0.0);
334-
}
335332
//
336333
// Step 3: Compute the row pointers for each group of IDs.
337334
//

0 commit comments

Comments
 (0)