Commit 06ff76b3 authored by Paul's avatar Paul
Browse files

Formatting

parent 6d1bb63a
...@@ -74,7 +74,7 @@ PYBIND11_MODULE(migraphx, m) ...@@ -74,7 +74,7 @@ PYBIND11_MODULE(migraphx, m)
m.def("parse_onnx", &migraphx::parse_onnx); m.def("parse_onnx", &migraphx::parse_onnx);
m.def("target", [](const std::string& name) -> migraphx::target { m.def("target", [](const std::string& name) -> migraphx::target {
if (name == "cpu") if(name == "cpu")
return migraphx::cpu::target{}; return migraphx::cpu::target{};
throw std::runtime_error("Target not found: " + name); throw std::runtime_error("Target not found: " + name);
}); });
......
...@@ -112,7 +112,8 @@ if(MIGRAPHX_ENABLE_GPU) ...@@ -112,7 +112,8 @@ if(MIGRAPHX_ENABLE_GPU)
endif() endif()
# Onnx test # Onnx test
add_executable(test_onnx onnx/onnx_test.cpp) set(TEST_ONNX_DIR ${CMAKE_CURRENT_SOURCE_DIR}/onnx)
add_executable(test_onnx ${TEST_ONNX_DIR}/onnx_test.cpp)
rocm_clang_tidy_check(test_onnx) rocm_clang_tidy_check(test_onnx)
target_link_libraries(test_onnx migraphx_onnx) target_link_libraries(test_onnx migraphx_onnx)
target_include_directories(test_onnx PUBLIC include) target_include_directories(test_onnx PUBLIC include)
......
...@@ -13,4 +13,4 @@ function(add_py_test NAME SCRIPT) ...@@ -13,4 +13,4 @@ function(add_py_test NAME SCRIPT)
COMMENT "${PYTHON_EXECUTABLE} ${SCRIPT}") COMMENT "${PYTHON_EXECUTABLE} ${SCRIPT}")
endfunction() endfunction()
add_py_test(test test.py) add_py_test(test test.py WORKING_DIRECTORY ${TEST_ONNX_DIR})
import migraphx import migraphx
p = migraphx.parse_onnx("conv_relu_maxpool.onnx")
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