Add annotations to deeplearning/fbgemm/fbgemm_gpu

Reviewed By: shannonzhu Differential Revision: D34217606 fbshipit-source-id: d9559ef75a85f1ad09d4c6c9349a4a53c447bcab
Compiler2 · Feb 14, 2022 · f4e0fcd · f4e0fcd
1 parent 1ddff63
commit f4e0fcd
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/fbgemm_gpu/bench/merge_embeddings_benchmark.py b/fbgemm_gpu/bench/merge_embeddings_benchmark.py
@@ -77,11 +77,11 @@ def benchmark(
     num_ads,
     embedding_dimension,
     ads_tables,
-    iters=10,
-    p2p_bw=False,
-    dst_device=0,
-    data_type="FP16",
-    include_quantization=False,
+    iters: int = 10,
+    p2p_bw: bool = False,
+    dst_device: int = 0,
+    data_type: str = "FP16",
+    include_quantization: bool = False,
 ) -> str:
     torch.cuda.set_device(dst_device)
     num_gpus = torch.cuda.device_count()