Commit 000b0b41 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon Committed by Robert Kimball

CPU Builder: Add double to the partial element type set (#1636)

* CPU Builder: Replace uint8_t with double in partial element type set

* Keep support for uint8_t kernels
parent 83a232ca
......@@ -171,6 +171,10 @@
{ \
PARTIAL_SELECT_RANK(KV, float, R, K); \
} \
else if (ET == element::f64) \
{ \
PARTIAL_SELECT_RANK(KV, double, R, K); \
} \
else if (ET == element::i8) \
{ \
PARTIAL_SELECT_RANK(KV, int8_t, R, K); \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment