Skip to content

GitLab

Commit 9e7e1253 authored by wsttiger's avatar wsttiger
Browse files

Added MIGRAPH_ENABLE_GPU on mnist and cifar10 targets

parent a5ee1886
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/onnx/CMakeLists.txt
View file @ 9e7e1253
...@@ -17,6 +17,7 @@ rocm_clang_tidy_check(read_onnx) ...@@ -17,6 +17,7 @@ rocm_clang_tidy_check(read_onnx)
target_link_libraries(read_onnx migraph_onnx) target_link_libraries(read_onnx migraph_onnx)
if(MIGRAPH_ENABLE_GPU)
add_executable(mnist mnist.cpp) add_executable(mnist mnist.cpp)
rocm_clang_tidy_check(mnist) rocm_clang_tidy_check(mnist)
target_link_libraries(mnist migraph_cpu migraph_gpu migraph_onnx) target_link_libraries(mnist migraph_cpu migraph_gpu migraph_onnx)
...@@ -25,7 +26,6 @@ add_executable(cifar10 cifar10.cpp) ...@@ -25,7 +26,6 @@ add_executable(cifar10 cifar10.cpp)
rocm_clang_tidy_check(cifar10) rocm_clang_tidy_check(cifar10)
target_link_libraries(cifar10 migraph_cpu migraph_gpu migraph_onnx) target_link_libraries(cifar10 migraph_cpu migraph_gpu migraph_onnx)
if(MIGRAPH_ENABLE_GPU)
add_executable(verify_onnx verify_onnx.cpp) add_executable(verify_onnx verify_onnx.cpp)
rocm_clang_tidy_check(verify_onnx) rocm_clang_tidy_check(verify_onnx)
target_link_libraries(verify_onnx migraph_onnx migraph_cpu migraph_gpu) target_link_libraries(verify_onnx migraph_onnx migraph_cpu migraph_gpu)
......
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