Add DRAM_SSD backend type to BackendType enum (#5802)

Summary:
Pull Request resolved: #5802

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

Reviewed By: EddyLXJ

Differential Revision: D106733255

fbshipit-source-id: 9fb1a5f57e801f03ac97785245e9094b8a1b475a

Author: Lizhe Ji

fbgemm_gpu/fbgemm_gpu/tbe/ssd/ssd_config.py

@@ -314,13 +314,15 @@ class BackendType(enum.IntEnum):
     SSD = 0
     DRAM = 1
     PS = 2
+    DRAM_SSD = 3
 
     @classmethod
     def from_str(cls, key: str) -> "BackendType":
         lookup = {
             "ssd": BackendType.SSD,
             "dram": BackendType.DRAM,
             "ps": BackendType.PS,
+            "dram_ssd": BackendType.DRAM_SSD,
         }
         if key in lookup:
             return lookup[key]

fbgemm_gpu/test/tbe/ssd/ssd_config_test.py

@@ -123,6 +123,17 @@ def test_dram(self) -> None:
 
         self.assertEqual(BackendType.from_str("dram"), BackendType.DRAM)
 
+    def test_ps(self) -> None:
+        from fbgemm_gpu.tbe.ssd import BackendType
+
+        self.assertEqual(BackendType.from_str("ps"), BackendType.PS)
+
+    def test_dram_ssd(self) -> None:
+        from fbgemm_gpu.tbe.ssd import BackendType
+
+        self.assertEqual(BackendType.from_str("dram_ssd"), BackendType.DRAM_SSD)
+        self.assertEqual(BackendType.DRAM_SSD.value, 3)
+
     def test_invalid_raises(self) -> None:
         from fbgemm_gpu.tbe.ssd import BackendType