Skip to content

Commit de248cb

Browse files
[npu] fix histogram bug (#1500)
1 parent aed827c commit de248cb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/npu/kernels/histogram_kernel.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ void HistogramKernel(const Context& dev_ctx,
2121
const phi::DenseTensor& input,
2222
const paddle::optional<phi::DenseTensor>& weight,
2323
int64_t bins,
24-
int min,
25-
int max,
24+
float min,
25+
float max,
2626
bool density,
2727
phi::DenseTensor* output) {
2828
PADDLE_ENFORCE_EQ(

0 commit comments

Comments
 (0)