DeepRec-AI/HybridBackend

EmbeddingLookupRewritingForDeepRecEV Add "part0" to op-name twice

Closed this issue · 0 comments

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