fix pca_numba_n

IntelPython · Nov 3, 2023 · 05cb6ba · 05cb6ba
1 parent 9751815
commit 05cb6ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpbench/benchmarks/pca/pca_numba_n.py b/dpbench/benchmarks/pca/pca_numba_n.py
@@ -39,7 +39,7 @@ def pca(data, dims_rescaled_data=2):
     evectors = evectors[:, :dims_rescaled_data]
 
     # carry out the transformation on the data using eigenvectors
-    tdata = np.dot(evectors.T, data.T).T
+    tdata = np.dot(evectors.T.astype(data.dtype), data.T).T
 
     # return the transformed data, eigenvalues, and eigenvectors
     return tdata, evalues, evectors