Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
06ff76b3
Commit
06ff76b3
authored
Feb 04, 2019
by
Paul
Browse files
Formatting
parent
6d1bb63a
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
6 additions
and
3 deletions
+6
-3
src/py/migraphx_py.cpp
src/py/migraphx_py.cpp
+1
-1
test/CMakeLists.txt
test/CMakeLists.txt
+2
-1
test/py/CMakeLists.txt
test/py/CMakeLists.txt
+1
-1
test/py/test.py
test/py/test.py
+2
-0
No files found.
src/py/migraphx_py.cpp
View file @
06ff76b3
...
...
@@ -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
);
});
...
...
test/CMakeLists.txt
View file @
06ff76b3
...
...
@@ -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
)
...
...
test/py/CMakeLists.txt
View file @
06ff76b3
...
...
@@ -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
}
)
test/py/test.py
View file @
06ff76b3
import
migraphx
p
=
migraphx
.
parse_onnx
(
"conv_relu_maxpool.onnx"
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment