Skip to content

Commit 9c70e47

Browse files
Lizhe Jifacebook-github-bot
authored andcommitted
Add DRAM_SSD backend type to BackendType enum
Summary: Add `DRAM_SSD` as a new backend type to the fbgemm `BackendType` enum in `fbgemm_gpu/tbe/ssd/ssd_config.py`. `DRAM_SSD` represents a composite DRAM + SSD KV backend for KVZCH virtual tables and is required by the `DRAM_SSD_VIRTUAL_TABLE` compute kernel, which references `BackendType.DRAM_SSD`. - Add `DRAM_SSD = 3` to `BackendType`. - Add `"dram_ssd"` mapping to `BackendType.from_str`. Differential Revision: D106733255
1 parent 6c71acd commit 9c70e47

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

fbgemm_gpu/fbgemm_gpu/tbe/ssd/ssd_config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -312,13 +312,15 @@ class BackendType(enum.IntEnum):
312312
SSD = 0
313313
DRAM = 1
314314
PS = 2
315+
DRAM_SSD = 3
315316

316317
@classmethod
317318
def from_str(cls, key: str) -> "BackendType":
318319
lookup = {
319320
"ssd": BackendType.SSD,
320321
"dram": BackendType.DRAM,
321322
"ps": BackendType.PS,
323+
"dram_ssd": BackendType.DRAM_SSD,
322324
}
323325
if key in lookup:
324326
return lookup[key]

0 commit comments

Comments
 (0)