Dropped unnecessary NDArray cast for potrf

apache · Feb 5, 2020 · 8fb0d1a · 8fb0d1a
1 parent 6946c65
commit 8fb0d1a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/benchmark/opperf/nd_operations/linalg_operators.py b/benchmark/opperf/nd_operations/linalg_operators.py
@@ -63,11 +63,11 @@ def run_linalg_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='nat
                                                   dtype=dtype,
                                                   ctx=ctx,
                                                   profiler=profiler,
-                                                  inputs=[{"A": mx.nd.array([[1, 0],
-                                                                             [0, 1]])},
-                                                          {"A": mx.nd.array([[2, -1, 0],
-                                                                             [-1, 2, -1],
-                                                                             [0, -1, 2]])}],
+                                                  inputs=[{"A": [[1, 0],
+                                                                 [0, 1]]},
+                                                          {"A": [[2, -1, 0],
+                                                                 [-1, 2, -1],
+                                                                 [0, -1, 2]]}],
                                                   warmup=warmup,
                                                   runs=runs)