Skip to content
This repository has been archived by the owner on Nov 25, 2024. It is now read-only.

Commit

Permalink
remove debug info
Browse files Browse the repository at this point in the history
  • Loading branch information
linhu-nv committed Jan 29, 2024
1 parent aa344fa commit e4f361a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cpp/src/wholememory_ops/functions/gather_scatter_func.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -367,7 +367,7 @@ void gather_temp_func(wholememory_gref_t embedding_gref,
int block_size = 1024;
int block_count = indice_count > 1568 ? 1568 : indice_count;
if (gather_sms != -1) block_count = gather_sms;
printf("the gather kernel sm number is %d\n", block_count);
// printf("the gather kernel sm number is %d\n", block_count);
kernel_fn<<<block_count, block_size, 0, stream>>>(embedding_gref,
embedding_desc,
static_cast<const IndexT*>(indices),
Expand Down

0 comments on commit e4f361a

Please sign in to comment.