minor changes

pytorch · Aug 26, 2022 · f86fe79 · f86fe79
1 parent 4c21738
commit f86fe79
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -341,12 +341,12 @@ jobs:
       - run_nvidia_smi
       - benchmark_layers_integration_test:
           device: "cuda"
-          layers: "groupnorm gsm_groupnorm instancenorm gsm_instancenorm layernorm gsm_layernorm"
+          layers: "groupnorm gsm_groupnorm instancenorm gsm_instancenorm layernorm gsm_layernorm dpmha"
           runtime_ratio_threshold: "2.5"
           memory_ratio_threshold: "1.6"
       - benchmark_layers_integration_test:
           device: "cuda"
-          layers: linear gsm_linear
+          layers: "linear gsm_linear"
           runtime_ratio_threshold: "3.6"
           memory_ratio_threshold: "13.0"
       - benchmark_layers_integration_test:

diff --git a/benchmarks/utils.py b/benchmarks/utils.py
@@ -225,6 +225,7 @@ def add_ratio(df, metric, variant):
                 df.loc[:, (metric, variant)] / df.loc[:, (metric, "control")]
             )
         else:
+            df[(metric, variant)] = np.nan
             df[(metric, f"{variant}/control")] = np.nan
 
     if "control" in results["variant"].tolist():