Skip to content

AttributeError: '_OpNamespace' 'fbgemm' object has no attribute 'get_bucket_sorted_indices_and_bucket_tensor' #5666

Description

@YINUO1221

When I run fbgemm_gpu/test/tbe/ssd/kv_backend_test.py in a GPU environment, I get the following error:

raise AttributeError(
AttributeError: '_OpNamespace' 'fbgemm' object has no attribute 'get_bucket_sorted_indices_and_bucket_tensor'
Is this because SSD TBE operators are not supported? However, I noticed that the torchrec code repository still uses the related functionality:

from fbgemm_gpu.tbe.ssd import SSDTableBatchedEmbeddingBags

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions