benchmark also supports double precision

springer13 · Aug 16, 2017 · 5d96590 · 5d96590
1 parent 67c0663
commit 5d96590
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/benchmark/python/benchmark.py b/benchmark/python/benchmark.py
@@ -29,6 +29,9 @@
     alpha = float(args.alpha)
 if( args.beta):
     beta = float(args.beta)
+if( args.floatType ):
+    if( args.floatType == 'double' ):
+        floatType = np.float64
 
 
 sizes = {}