test_cpu.py 400 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)
Paul's avatar
Paul committed
5
s1 = p.get_shape()
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)
Paul's avatar
Paul committed
9
10
s2 = p.get_shape()
assert s1 == s2
Paul's avatar
Paul committed
11
12
params = {}
for key, value in p.get_parameter_shapes().items():
Paul's avatar
Paul committed
13
    print("Parameter {} -> {}".format(key, value))
Paul's avatar
Paul committed
14
15
16
17
    params[key] = migraphx.generate_argument(value)

r = p.run(params)
print(r)