Arm backend: Use correct name when indexing state_dict

[Sebastian-Larsson]

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/12954

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 2 New Failures, 1 Unrelated Failure

As of commit fd9f7a5 with merge base b0dda93 ():

NEW FAILURES - The following jobs have failed:

• pull / test-models-linux (phi_4_mini, portable, linux.4xlarge.memory) / linux-job (gh)
  RuntimeError: Command docker exec -t 182c3847bd1f1e83b733eefa895a35e2c50415cf565479f6294b652ba7415043 /exec failed with exit code 1
• pull / unittest-editable / macos / macos-job (gh)
  backends/xnnpack/test/ops/test_conv1d.py::TestConv1d::test_qs8_conv1d_batchnorm_seq

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• trunk / test-models-macos-coreml (emformer_transcribe) / macos-job (gh) (trunk failure)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 1

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[Sebastian-Larsson]

CI failures seem unrelated

[digantdesai]

nit make a util def get_name(self, node)?

[Sebastian-Larsson]

Local to this file? Seems a little redundant to me, I suppose that would look something like this

def get_name(self, node):
     if is_buffer(self.exported_program, node):
         return self.exported_program.graph_signature.inputs_to_buffers[node.name]
     # Reapeat for param etc.
     ...

And this is done 2 times only.

I think it would belong in torch._export.utils like is_buffer etc.

[digantdesai]

even local would make it cleaner. Feel free to put up a PR against _export.utils if you feel like others are doing this or should do it this way.