"vscode:/vscode.git/clone" did not exist on "c62d7bf92d6c29b78d2fc87fc6c12a7244340b3d"
test_cpu.py 429 Bytes
Newer Older
Paul's avatar
Paul committed
1
import migraphx
Paul's avatar
Paul committed
2

Khalique's avatar
Khalique committed
3
p = migraphx.parse_onnx("conv_relu_maxpool_test.onnx")
Paul's avatar
Paul committed
4
print(p)
5
s1 = p.get_output_shapes()[-1]
Paul's avatar
Paul committed
6
print("Compiling ...")
Paul's avatar
Paul committed
7
p.compile(migraphx.get_target("cpu"))
Paul's avatar
Paul committed
8
print(p)
9
s2 = p.get_output_shapes()[-1]
Paul's avatar
Paul committed
10
assert s1 == s2
Paul's avatar
Paul committed
11
params = {}
kahmed10's avatar
kahmed10 committed
12

Paul's avatar
Paul committed
13
for key, value in p.get_parameter_shapes().items():
Paul's avatar
Paul committed
14
    print("Parameter {} -> {}".format(key, value))
Paul's avatar
Paul committed
15
16
    params[key] = migraphx.generate_argument(value)

17
r = p.run(params)[-1]
Paul's avatar
Paul committed
18
print(r)