fix build error

282bd488 · Shucai Xiao · 88ed7f85 · 282bd488
Commit 282bd488 authored Jun 28, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/targets/cpu/lowering.cpp src/targets/cpu/lowering.cpp +2 -0

No files found.
--- a/src/targets/cpu/lowering.cpp
+++ b/src/targets/cpu/lowering.cpp
@@ -13,6 +13,8 @@
 #include <migraphx/op/pad.hpp>
 #include <migraphx/op/pooling.hpp>
 #include <migraphx/op/softmax.hpp>
+#include <migraphx/op/argmax.hpp>
+#include <migraphx/op/argmin.hpp>
 #include <migraphx/shape_for_each.hpp>
 #include <migraphx/iterator_for.hpp>
 #include <migraphx/par_dfor.hpp>