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 a992aa5 commit ddc026bCopy full SHA for ddc026b
graph_net/torch/test_compiler.py
@@ -124,8 +124,8 @@ def get_input_dict(args):
124
inputs_params = utils.load_converted_from_text(f"{args.model_path}")
125
params = inputs_params["weight_info"]
126
for tensor_meta in params.values():
127
- if hasattr(tensor_meta, "device"):
128
- tensor_meta.device = args.device
+ if "device" in tensor_meta["info"]:
+ tensor_meta["info"]["device"] = args.device
129
return {
130
k: utils.replay_tensor(v).to(torch.device(args.device))
131
for k, v in params.items()
0 commit comments