Skip to content
Snippets Groups Projects
Commit a4ed5700 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Driver] small fix for handling single output

parent c6b103a6
No related branches found
No related tags found
No related merge requests found
...@@ -119,6 +119,8 @@ if __name__ == "__main__": ...@@ -119,6 +119,8 @@ if __name__ == "__main__":
for ifn in inputfile: for ifn in inputfile:
ibuf_normal.append(np.load(ifn)) ibuf_normal.append(np.load(ifn))
obuf_normal = accel.execute(ibuf_normal) obuf_normal = accel.execute(ibuf_normal)
if not isinstance(obuf_normal, list):
obuf_normal = [obuf_normal]
for o, obuf in enumerate(obuf_normal): for o, obuf in enumerate(obuf_normal):
np.save(outputfile[o], obuf) np.save(outputfile[o], obuf)
elif exec_mode == "throughput_test": elif exec_mode == "throughput_test":
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment