未验证 提交 48090c72 编写于 作者: H hellockx 提交者: GitHub

add register of bincount (#51508)

* Update interpreter_util.cc

* Update bincount_kernel.cc

* Update bincount_kernel.cu
上级 5afab2cd
......@@ -57,7 +57,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
"arg_sort",
"atan2",
"auc",
"bincount",
"clip_by_norm",
"complex",
"conv3d_coo",
......
......@@ -111,4 +111,6 @@ PD_REGISTER_KERNEL(bincount,
float,
double,
int,
int64_t) {}
int64_t) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
......@@ -158,4 +158,6 @@ PD_REGISTER_KERNEL(bincount,
float,
double,
int,
int64_t) {}
int64_t) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册