Skip to content

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

@YINUO1221

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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions