Browse Source

!3187 Remove useless log in TopK kernel

Merge pull request !3187 from chenweifeng/topk
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b5d8dad47d
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/topk_gpu_kernel.h

+ 0
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/topk_gpu_kernel.h View File

@@ -51,7 +51,6 @@ class TopKGpuKernel : public GpuKernel {
reinterpret_cast<cudaStream_t>(stream_ptr));

if (sorted_ == false) {
std::cout << "================BitonicSortByKey" << std::endl;
BitonicSortByKey(outer_size_, k_, output_addr, indices, data_buff, index_buff,
reinterpret_cast<cudaStream_t>(stream_ptr));
}


Loading…
Cancel
Save