"docs/archive_en_US/TrialExample/Trials.md" did not exist on "bc0e55a00bbbc825f27d851ccc58a749d18b4fd9"
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():
else:
test_input = np.zeros(in_shape).astype(get_np_datatype(in_type))
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:
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:
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