Skip to content

Commit e75f868

Browse files
MoFHekarhdong
authored andcommitted
[Fix] Prefix bit should be in the upper bit.
1 parent e1be08e commit e75f868

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -419,7 +419,7 @@ def call(self, features):
419419
boundaries=fea_info['boundaries'])(input_tensor)
420420
# To prepare for GPU table combined queries, use a prefix to distinguish different features in a table.
421421
if fea_info['ptype'] == 'user_occupation_label':
422-
input_tensor_prefix_code = int(fea_info['code']) << 1
422+
input_tensor_prefix_code = int(fea_info['code']) << 48
423423
else:
424424
input_tensor_prefix_code = None
425425
if input_tensor_prefix_code is not None:

0 commit comments

Comments
 (0)