Skip to content

Commit

Permalink
[fix] Change the parameter max_capacity sent by the cuckoo gpu api to…
Browse files Browse the repository at this point in the history
… hkv to sys.maxsize (#382)
  • Loading branch information
LinGeLin authored Feb 17, 2024
1 parent 017d05e commit 8b071de
Showing 1 changed file with 1 addition and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -103,9 +103,7 @@ def __init__(
self._init_size = init_size
self._name = name
self._new_obj_trackable = None # for restore op can easily found this table
self._max_capacity = init_size
if self._max_capacity == 0:
self._max_capacity = 1024 * 1024
self._max_capacity = sys.maxsize
self._max_hbm_for_values = sys.maxsize
self._device_type = tf_device.DeviceSpec.from_string(
self._device).device_type
Expand Down

0 comments on commit 8b071de

Please sign in to comment.