fix

apache · Mar 23, 2020 · e37c1cb · e37c1cb
1 parent a5d55de
commit e37c1cb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/benchmark/opperf/nd_operations/array_manipulation_operators.py b/benchmark/opperf/nd_operations/array_manipulation_operators.py
@@ -97,7 +97,7 @@ def run_rearrange_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='
     mx_rearrange_ops = get_all_rearrange_operators()
 
     # Run benchmarks
-    mx_rearrange_op_results = run_op_benchmarks(mx_rearrange_ops, dtype, ctx, profiler, warmup, runs)
+    mx_rearrange_op_results = run_op_benchmarks(mx_rearrange_ops, dtype, ctx, profiler, int64_tensor, warmup, runs)
     return mx_rearrange_op_results
 
 
@@ -129,7 +129,7 @@ def run_shape_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='nati
     mx_shape_ops = get_all_shape_operators()
 
     # Run benchmarks
-    mx_shape_op_results = run_op_benchmarks(mx_shape_ops, dtype, ctx, profiler, warmup, runs)
+    mx_shape_op_results = run_op_benchmarks(mx_shape_ops, dtype, ctx, profiler, int64_tensor, warmup, runs)
     return mx_shape_op_results
 
 
@@ -161,7 +161,7 @@ def run_expanding_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='
     mx_expanding_ops = get_all_expanding_operators()
 
     # Run benchmarks
-    mx_expanding_op_results = run_op_benchmarks(mx_expanding_ops, dtype, ctx, profiler, warmup, runs)
+    mx_expanding_op_results = run_op_benchmarks(mx_expanding_ops, dtype, ctx, profiler, int64_tensor, warmup, runs)
     return mx_expanding_op_results
 
 
@@ -193,7 +193,7 @@ def run_rounding_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='n
     mx_rounding_ops = get_all_rounding_operators()
 
     # Run benchmarks
-    mx_rounding_op_results = run_op_benchmarks(mx_rounding_ops, dtype, ctx, profiler, warmup, runs)
+    mx_rounding_op_results = run_op_benchmarks(mx_rounding_ops, dtype, ctx, profiler, int64_tensor, warmup, runs)
     return mx_rounding_op_results
 
 

diff --git a/benchmark/opperf/rules/default_params.py b/benchmark/opperf/rules/default_params.py
@@ -603,7 +603,6 @@
                                 "p": DEFAULT_P,
                                 "k_nd": DEFAULT_K_ND_LARGE_TENSOR,
                                 "p_nd": DEFAULT_P_ND_LARGE_TENSOR,
-                                "axis_shape": DEFAULT_AXIS_SHAPE,
                                 "axis": DEFAULT_AXIS,
                                 "weight" : DEFAULT_WEIGHT_LARGE_TENSOR,
                                 "weight32" : DEFAULT_WEIGHT_LARGE_TENSOR,