Skip to content

Commit cd73174

Browse files
Lizhe Jimeta-codesync[bot]
authored andcommitted
Add BackendType.DRAM_SSD enum value and DRAM_SSD_VIRTUAL_TABLE enum to torchrec and sparsenn_configs (#5774)
Summary: Pull Request resolved: #5774 X-link: https://github.com/facebookresearch/FBGEMM/pull/2702 X-link: meta-pytorch/torchrec#4273 Introduces the `DRAM_SSD` backend type to support a composite DRAM and SSD storage hierarchy for virtual embedding tables. This expands the available backend storage options to enable multi-tier caching configurations in TorchRec and FBGEMM. Reviewed By: EddyLXJ Differential Revision: D105646650 fbshipit-source-id: 71d4b1e2e56c7c7798acd076dd7eb0f7bf38197a
1 parent 9ed8ef5 commit cd73174

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_common.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,12 +336,14 @@ class BackendType(enum.IntEnum):
336336
SSD = 0
337337
DRAM = 1
338338
PS = 2
339+
DRAM_SSD = 3
339340

340341
@classmethod
341342
def from_str(cls, key: str) -> "BackendType":
342343
lookup = {
343344
"ssd": BackendType.SSD,
344345
"dram": BackendType.DRAM,
346+
"dram_ssd": BackendType.DRAM_SSD,
345347
}
346348
if key in lookup:
347349
return lookup[key]

0 commit comments

Comments
 (0)