try

Author: ChaiBapchya

benchmark/opperf/utils/benchmark_utils.py

@@ -56,7 +56,7 @@ def _run_nd_operator_performance_test(op, inputs, run_backward, warmup, runs, ar
         benchmark_helper_func = nd_forward_and_profile
 
     if not args_list:
-        _, _ = benchmark_helper_func(op, warmup, None, **kwargs_list[0])
+        _, _ = benchmark_helper_func(op, warmup, [], **kwargs_list[0])
     else:
     # Warm up, ignore the profiler output
         _, _ = benchmark_helper_func(op, warmup, args_list[0], **kwargs_list[0])
@@ -66,7 +66,7 @@ def _run_nd_operator_performance_test(op, inputs, run_backward, warmup, runs, ar
     logging.info("Begin Benchmark - {name}".format(name=op.__name__))
     if not args_list:
         for idx, kwargs in enumerate(kwargs_list):
-            _, profiler_output = benchmark_helper_func(op, runs, None, **kwargs)
+            _, profiler_output = benchmark_helper_func(op, runs, [], **kwargs)
 
             # Add inputs used for profiling this operator into result
             profiler_output["inputs"] = inputs[idx]

benchmark/opperf/utils/ndarray_utils.py

@@ -47,7 +47,7 @@ def nd_forward_backward_and_profile(op, runs, *args, **kwargs):
     """
     for _ in range(runs):
         with mx.autograd.record():
-            if(args==(None,)):
+            if not isinstance(args[0],nd.NDArray):
                 res = op(**kwargs)
             else:
                 res = op(*args, **kwargs)
@@ -79,7 +79,7 @@ def nd_forward_and_profile(op, runs, *args, **kwargs):
     any results from NDArray operation execution
     """
     for _ in range(runs):
-        if(args==(None,)):
+        if not isinstance(args[0],nd.NDArray):
             res = op(**kwargs)
         else:
             res = op(*args, **kwargs)