Added remaining misc ops, including Custom op logic

apache · Feb 6, 2020 · a721938 · a721938
1 parent 7b00a70
commit a721938
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/benchmark/opperf/rules/default_params.py b/benchmark/opperf/rules/default_params.py
@@ -239,4 +239,4 @@
                           "weight", "weight32", "grad", "mean", "var", "mom", "n", "d",
                           "v", "z", "g", "delta", "args", "indices", "shape_like", "y",
                           "x", "condition", "a", "index", "raveL_data", "label", "lrs",
-                          "wds", "weights_sum_sq", "grads_sum_sq"]
+                          "wds", "weights_sum_sq", "grads_sum_sq", "mhs"]
diff --git a/benchmark/opperf/utils/op_registry_utils.py b/benchmark/opperf/utils/op_registry_utils.py
@@ -423,7 +423,7 @@ def get_all_miscellaneous_operators():
     -------
     {"operator_name": {"has_backward", "nd_op_handle", "params"}}
     """
-    misc_ops = ['squeeze', 'all_finite', 'clip', 'multi_lars', 'SequenceReverse', 'SequenceLast', 'SequenceMask']
+    misc_ops = ['squeeze', 'all_finite', 'clip', 'multi_lars', 'SequenceReverse', 'SequenceLast', 'SequenceMask', 'cast_storage', 'cumsum', 'fill_element_0index']
 
     # Get all mxnet operators
     mx_operators = _get_all_mxnet_operators()