You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: packages/fbgemm/README.md
+11-8Lines changed: 11 additions & 8 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,9 +1,14 @@
1
-
# FBGEMM Embedding Benchmark
1
+
# FBGEMM Benchmarks
2
+
2
3
`fbgemm_embedding` is a benchmark that simulates embedding table lookups representative of Model A and Model B workloads.
3
4
The benchmark closely mimics the hotspot functions in table-based embedding (TBE) inference operations, which are commonly used in recommendation systems and deep learning models.
4
5
6
+
`fbgemm_cpu_fp16` is a microbenchmark that evaluates the FP16-based matrix multiplication performance, which are common in Meta's recommendation and ranking related workloads.
7
+
8
+
`fbgemm_cpu_spmdm8` is a C++ based micro-benchmark with some simple parameters and it benchmarks the TBE kernel. It's used for quick prototyping and performance validation.
9
+
5
10
## Installation
6
-
To install the FBGEMM embedding benchmark, execute one of the following command:
11
+
To install the FBGEMM benchmarks, execute one of the following command:
7
12
```bash
8
13
./benchpress -b ai install fbgemm_embedding_a_single
9
14
./benchpress -b ai install fbgemm_embedding_a_spec
@@ -14,7 +19,7 @@ To install the FBGEMM embedding benchmark, execute one of the following command:
14
19
./benchpress -b ai install fbgemm_cpu_spmdm8
15
20
```
16
21
17
-
## Run FBGEMM Embedding Benchmark
22
+
## Run FBGEMM Benchmarks
18
23
### Job - `fbgemm_embedding_a_single`
19
24
`fbgemm_embedding_a_single` simulates the Model A workload using a single representative embedding table.
20
25
This benchmark is designed to evaluate performance with a simplified embedding configuration.
@@ -29,14 +34,12 @@ This benchmark is designed to evaluate performance with a simplified embedding c
29
34
`fbgemm_embedding_b_spec_int8` simulates the Model B workload using a representative set of embedding tables with int8 quantized precision.
30
35
31
36
### Job - `fbgemm_cpu_fp16`
32
-
`fbgemm_cpu_a` executes the FP16Benchmark from FBGEMM-CPU.
37
+
`fbgemm_cpu_a` executes the `FP16Benchmark` from FBGEMM.
33
38
34
39
### Job - `fbgemm_embedding_cpu_b`
35
-
`fbgemm_cpu_b` executes the EmbeddingSpMDM8BitBenchmark from FBGEMM-CPU.
36
-
This is a C++ based micro-benchmark which stresses the TBE kernel.
37
-
This benchmark is used for quick prototyping and performance validation.
40
+
`fbgemm_cpu_b` executes the `EmbeddingSpMDM8BitBenchmark` from FBGEMM.
38
41
39
-
To run the FBGEMM embedding benchmarks, please use the following commands:
42
+
To run the FBGEMM benchmarks, please use the following commands:
0 commit comments