"...composable_kernel_onnxruntime.git" did not exist on "15c89e81f0587e8b46caa6062040c469a97ebc09"
Commit 454d9c08 authored by Umang Yadav's avatar Umang Yadav
Browse files

Merge branch 'develop' into resnet50_partition

parents 1a92cca5 6e86734d
...@@ -220,10 +220,16 @@ def main(): ...@@ -220,10 +220,16 @@ def main():
else: else:
test_input = np.zeros(in_shape).astype(get_np_datatype(in_type)) test_input = np.zeros(in_shape).astype(get_np_datatype(in_type))
test_inputs[name] = test_input test_inputs[name] = test_input
params[name] = migraphx.argument(test_input) migraphx_arg = migraphx.argument(test_input)
if not args.offload_copy:
migraphx_arg = migraphx.to_gpu(migraphx_arg)
params[name] = migraphx_arg
if not args.ort_run: if not args.ort_run:
pred_migx = np.array(model.run(params)[-1]) if not args.offload_copy:
pred_migx = np.array(migraphx.from_gpu(model.run(params)[-1]))
else:
pred_migx = np.array(model.run(params)[-1])
if use_onnx: if use_onnx:
sess_op = ort.SessionOptions() sess_op = ort.SessionOptions()
......
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