fix hashtable_inl.h (PaddlePaddle#34)

zmxdream · Jul 1, 2022 · eb10366 · eb10366
1 parent b9c6741
commit eb10366
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/framework/fleet/heter_ps/hashtable_inl.h b/paddle/fluid/framework/fleet/heter_ps/hashtable_inl.h
@@ -101,7 +101,7 @@ __global__ void dy_mf_search_kernel(Table* table,
         *(uint64_t*)(cur_p + k * 4) = *(uint64_t*)(input_p + k * 4);
       }
       else {
-        int len_per_thread = (len - 9) / (blockDim.x - 9);
+        int len_per_thread = (len - 9) / (blockDim.y - 9);
         int remain = (len - 9) % (blockDim.y - 9);
         int real_len = len_per_thread;
         if ((k - 9) < remain) real_len++;
@@ -116,7 +116,7 @@ __global__ void dy_mf_search_kernel(Table* table,
         for(int j = left; j < right; j++) *(float*)(cur_p + (j + 1) * 4) = *(float*)(input_p + (j + 1) * 4);
       }
     } else {
-      if (keys[i] != 0) printf("pull miss key: %d",keys[i]);
+      if (keys[i] != 0) printf("pull miss key: %llu",keys[i]);
     }
   }
 }