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
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