diff --git a/oneflow/user/kernels/acc_kernel.cpp b/oneflow/user/kernels/acc_kernel.cpp index 6d0f82dd4d0..97f31fd5dbc 100644 --- a/oneflow/user/kernels/acc_kernel.cpp +++ b/oneflow/user/kernels/acc_kernel.cpp @@ -36,6 +36,7 @@ class AccKernel final : public user_op::OpKernel { std::unique_ptr primitive = ep::primitive::NewPrimitive(ctx->device_type(), in->data_type()); CHECK(primitive); + std::cout << ctx->Tensor4ArgNameAndIndex("in", 0)->shape_view().elem_cnt() << std::endl; primitive->Launch(ctx->stream(), out->dptr(), in->dptr(), out->mut_dptr(), in->shape_view().elem_cnt()); }