We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 74d19b7 commit 9dee9ebCopy full SHA for 9dee9eb
graph_net/torch/test_compiler.py
@@ -126,8 +126,8 @@ def get_input_dict(args):
126
inputs_params = utils.load_converted_from_text(f"{args.model_path}")
127
params = inputs_params["weight_info"]
128
for tensor_meta in params.values():
129
- if hasattr(tensor_meta, "device"):
130
- tensor_meta.device = args.device
+ if "device" in tensor_meta["info"]:
+ tensor_meta["info"]["device"] = args.device
131
return {
132
k: utils.replay_tensor(v).to(torch.device(args.device))
133
for k, v in params.items()
0 commit comments