Unverified Commit 6e86734d authored by kahmed10's avatar kahmed10 Committed by GitHub
Browse files

update script when offload copy is disabled (#2348)

parent c3990622
...@@ -220,9 +220,15 @@ def main(): ...@@ -220,9 +220,15 @@ 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:
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]) pred_migx = np.array(model.run(params)[-1])
if use_onnx: if use_onnx:
......
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