提交 7f45b951 编写于 作者: M minqiyang

Polish code

上级 68a07328
......@@ -29,6 +29,7 @@ DECLARE_bool(benchmark);
DEFINE_bool(check_nan_inf, false,
"Checking whether operator produce NAN/INF or not. It will be "
"extremely slow so please use this flag wisely.");
DEFINE_int32(inner_op_parallelism, 0, "number of threads for inner op");
namespace paddle {
namespace framework {
......
......@@ -45,7 +45,7 @@ class HashKerel : public framework::OpKernel<T> {
for (int idx = 0; idx < seq_length; ++idx) {
for (int ihash = 0; ihash != num_hash; ++ihash) {
output[idx * num_hash + ihash] =
XXH64(input, sizeof(int) * last_dim, ihash) % mod_by;
XXH32(input, sizeof(int) * last_dim, ihash) % mod_by;
}
input += last_dim;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册