Commit 07c8f0f
authored
Arm backend: Change node.name to buffer name when indexing state_dict (pytorch#12485)
Using node.name for indexing state_dict can result in the following:
`KeyError: 'b__tensor_constant1'`.
Using the following to fetch the buffer name is the correct way to do
it:
```
buffer_name = self.exported_program.graph_signature.inputs_to_buffers[
node.name
]
```
Signed-off-by: Sebastian Larsson <[email protected]>1 parent 5128307 commit 07c8f0f
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
47 | 47 | | |
48 | 48 | | |
49 | 49 | | |
50 | | - | |
| 50 | + | |
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
| |||
0 commit comments