Skip to content

Add DRAM_SSD backend type to BackendType enum#5802

Open
lizhe-ji wants to merge 1 commit into
pytorch:mainfrom
lizhe-ji:export-D106733255
Open

Add DRAM_SSD backend type to BackendType enum#5802
lizhe-ji wants to merge 1 commit into
pytorch:mainfrom
lizhe-ji:export-D106733255

Conversation

@lizhe-ji
Copy link
Copy Markdown

@lizhe-ji lizhe-ji commented May 29, 2026

Summary:
X-link: https://github.com/facebookresearch/FBGEMM/pull/2730

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

Summary:
X-link: facebookresearch/FBGEMM#2730

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
@lizhe-ji lizhe-ji force-pushed the export-D106733255 branch from 9c70e47 to bf74abf Compare June 1, 2026 19:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant