benchdnn: ip: add DLRM training sizes

aaraujom · aaraujom · commit 07579e6c0c68 · 2020-03-31T09:33:27.000-07:00
diff --git a/tests/benchdnn/inputs/ip/ip_dlrm b/tests/benchdnn/inputs/ip/ip_dlrm
@@ -0,0 +1,8 @@
+ic13oc512n"DLRM:0"
+ic512oc256n"DLRM:1"
+ic256oc128n"DLRM:2"
+ic479oc1024n"DLRM:3"
+ic1024oc1024n"DLRM:4"
+ic1024oc512n"DLRM:5"
+ic512oc256n"DLRM:6"
+ic256oc1n"DLRM:7"
diff --git a/tests/benchdnn/inputs/ip/perf_ip_training b/tests/benchdnn/inputs/ip/perf_ip_training
@@ -12,3 +12,4 @@
 --mb=2048 --batch=ip_ncf
 --mb=1024 --batch=ip_alexnet
 --mb=512 --batch=ip_maskrcnn
+--mb=2048 --batch=ip_dlrm
