Skip to content

Commit

Permalink
refine
Browse files Browse the repository at this point in the history
  • Loading branch information
marigoold committed Jul 28, 2023
1 parent e0899c4 commit 466f175
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions oneflow/user/kernels/bincount_kernel.cu
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,6 @@ __global__ static void BinCountCompute(const IDX* in_ptr, T* out_ptr, int64_t si
} else {
__shared__ T shm[kCudaThreadsNumPerBlock];
T zero = GetZeroVal<T>();
// printf("tid: %" PRId64 " , size is %" PRId64 "\n", threadIdx.x, size);
shm[threadIdx.x] = zero;
IDX idx = 0;
CUDA_1D_KERNEL_LOOP(i, size) { idx = *(in_ptr + i); }
Expand Down Expand Up @@ -91,7 +90,6 @@ class CUDABinCountKernel final : public user_op::OpKernel {
void Compute(user_op::KernelComputeContext* ctx) const override {
const user_op::Tensor* in = ctx->Tensor4ArgNameAndIndex("in", 0);
size_t out_size = ctx->Attr<int64_t>("size");
;
user_op::Tensor* out = ctx->Tensor4ArgNameAndIndex("out", 0);
const IDX* in_ptr = in->dptr<IDX>();
T* out_ptr = out->mut_dptr<T>();
Expand Down

0 comments on commit 466f175

Please sign in to comment.