Skip to content

Commit 0902ba6

Browse files
committed
fix[hardware_predict]: Force x array to float64
1 parent 2492ce9 commit 0902ba6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/backends/vitis_accelerator/vitis_accelerator_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def hardware_predict(self, model, x, target="hw", debug=False, profilingRepeat=-
138138
elif method == "lib":
139139
"""Run the hardware prediction using a shared library."""
140140
# Set array to contiguous memory layout
141-
X_test = np.ascontiguousarray(x)
141+
X_test = np.ascontiguousarray(x, dtype=np.float64)
142142

143143
# Create prediction array
144144
config = VitisAcceleratorConfig(model.config)

0 commit comments

Comments
 (0)