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

Formatting

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