EmbeddingLookupRewritingForDeepRecEV Add "part0" to op-name twice
Closed this issue · 0 comments
karterotte commented
Current behavior
One node name should be "XX/embedding_weights/part_0",but now it is "XX/embedding_weights/part_0/part_0"
Expected behavior
Remove EmbeddingLookupRewritingForDeepRecEV.build_unsharded_weights rename func like
if name.endswith('/part_0'):
return fn(name, *args, **kwargs)
else:
return fn(f'{name}/part_0', *args, **kwargs)
It should support mutilple partition like "part_1X"
Willing to contribute
Yes