Skip to content

Commit 8b071de

Browse files
authored
[fix] Change the parameter max_capacity sent by the cuckoo gpu api to hkv to sys.maxsize (#382)
1 parent 017d05e commit 8b071de

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/ops/cuckoo_hashtable_ops.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -103,9 +103,7 @@ def __init__(
103103
self._init_size = init_size
104104
self._name = name
105105
self._new_obj_trackable = None # for restore op can easily found this table
106-
self._max_capacity = init_size
107-
if self._max_capacity == 0:
108-
self._max_capacity = 1024 * 1024
106+
self._max_capacity = sys.maxsize
109107
self._max_hbm_for_values = sys.maxsize
110108
self._device_type = tf_device.DeviceSpec.from_string(
111109
self._device).device_type

0 commit comments

Comments
 (0)