Skip to content

Commit ee41cba

Browse files
Fix problem with float32 json dump (#112)
* fix problem with float32 json dump * less complexity * pep8
1 parent df83b58 commit ee41cba

File tree

3 files changed

+8
-4
lines changed

3 files changed

+8
-4
lines changed

Diff for: bench.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -540,11 +540,15 @@ def print_output(library, algorithm, stages, params, functions,
540540
data[i], alg_instance, alg_params)
541541
result.update({'time[s]': times[i]})
542542
if isinstance(metric_type, str):
543-
result.update({f'{metric_type}': metrics[i]})
543+
result.update({f'{metric_type}': np.float64(metrics[i])
544+
if isinstance(metrics[i], np.float32)
545+
else metrics[i]})
544546
elif isinstance(metric_type, list):
545547
for ind, val in enumerate(metric_type):
546548
if metrics[ind][i] is not None:
547-
result.update({f'{val}': metrics[ind][i]})
549+
result.update({f'{val}': np.float64(metrics[ind][i])
550+
if isinstance(metrics[ind][i], np.float32)
551+
else metrics[ind][i]})
548552
if hasattr(params, 'n_classes'):
549553
result['input_data'].update({'classes': params.n_classes})
550554
if hasattr(params, 'n_clusters'):

Diff for: configs/xpu/kmeans.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"algorithm": "kmeans",
55
"data-format": "pandas",
66
"data-order": "C",
7-
"dtype": ["float64"],
7+
"dtype": ["float32", "float64"],
88
"device": ["host", "cpu", "gpu", "none"]
99
},
1010
"cases": [

Diff for: configs/xpu/pca.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"algorithm": "pca",
55
"data-format": "pandas",
66
"data-order": "F",
7-
"dtype": ["float64"],
7+
"dtype": ["float32", "float64"],
88
"device": ["host", "cpu", "none"]
99
},
1010
"cases": [

0 commit comments

Comments
 (0)