Abstract:High-performance flash-based key-value stores in data-centers utilize large amounts of DRAM to cache hot data. However, motivated by the high cost and power consumption of DRAM, server designs with lower DRAM-per-compute ratio are becoming popular. These low-cost servers enable scale-out services by reducing server workload densities. This results in improvements to overall service reliability, leading to a decrease in the total cost of ownership (TCO) for scalable workloads. Nevertheless, for key-value stores with large memory footprints, these reduced DRAM servers degrade performance due to an increase in both IO utilization and data access latency. In this scenario, a standard practice to improve performance for sharded databases is to reduce the number of shards per machine, which degrades the TCO benefits of reduced DRAM low-cost servers. In this work, we explore a practical solution to improve performance and reduce the costs and power consumption of key-value stores running on DRAM-constrained servers by using Storage Class Memories (SCM). SCMs in a DIMM form factor, although slower than DRAM, are sufficiently faster than flash when serving as a large extension to DRAM. With new technologies like Compute Express Link, we can expand the memory capacity of servers with high bandwidth and low latency connectivity with SCM. In this article, we use Intel Optane PMem 100 Series SCMs (DCPMM) in AppDirect mode to extend the available memory of our existing single-socket platform deployment of RocksDB (one of the largest key-value stores at Meta). We first designed a hybrid cache in RocksDB to harness both DRAM and SCM hierarchically. We then characterized the performance of the hybrid cache for three of the largest RocksDB use cases at Meta (ChatApp, BLOB Metadata, and Hive Cache). Our results demonstrate that we can achieve up to 80% improvement in throughput and 20% improvement in P95 latency over the existing small DRAM single-socket platform, while maintaining a 43–48% cost improvement over our large DRAM dual-socket platform. To the best of our knowledge, this is the first study of the DCPMM platform in a commercial data center.

Supporting Massive DLRM Inference Through Software Defined Memory

MTrainS: Improving DLRM training efficiency using heterogeneous memories

RecNMP: Accelerating Personalized Recommendation with Near-Memory Processing

Improving Performance of Flash Based Key-Value Stores Using Storage Class Memory As a Volatile Memory Extension

UpDLRM: Accelerating Personalized Recommendation using Real-World PIM Architecture

RecPIM: Efficient In-Memory Processing for Personalized Recommendation Inference Using Near-Bank Architecture

LLM in a flash: Efficient Large Language Model Inference with Limited Memory

RecSSD: near data processing for solid state drive based recommendation inference

DLRover-RM: Resource Optimization for Deep Recommendation Models Training in the Cloud

Power-optimized Deployment of Key-value Stores Using Storage Class Memory

NDRec: A Near-Data Processing System for Training Large-Scale Recommendation Models

PIMPR: PIM-based Personalized Recommendation with Heterogeneous Memory Hierarchy

An Efficient Near-Bank Processing Architecture for Personalized Recommendation System

Near-Memory Processing in Action: Accelerating Personalized Recommendation with AxDIMM.

Retention-Aware Hybrid Main Memory (RAHMM): Big DRAM and Little SCM

Accelerating Bandwidth-Bound Deep Learning Inference with Main-Memory Accelerators

Harnessing Your DRAM and SSD for Sustainable and Accessible LLM Inference with Mixed-Precision and Multi-level Caching

A Survey of Resource Management for Processing-in-Memory and Near-Memory Processing Architectures

MARM: Unlocking the Future of Recommendation Systems through Memory Augmentation and Scalable Complexity

Software-Defined Far Memory in Warehouse-Scale Computers