Make type of logits a template parameter

[sxu]

Summary: We use fp16 with CoreML.

Differential Revision: D84211619

[pytorch-bot]

🔗 Helpful Links

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

• 📄 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 Cancelled Job

As of commit df43af5 with merge base a26412e ():

NEW FAILURES - The following jobs have failed:

• pull / test-samsung-models-linux / linux-job (gh)
  RuntimeError: Command docker exec -t 57b1b97f0612939fc1d3e3dfc2030ce38126b1bd81f4080c7efb9697b27fb376 /exec failed with exit code 1
• Test CUDA Builds / test-voxtral-cuda-e2e / linux-job (gh)
  RuntimeError: Command docker exec -t bae0833f1039ee2a270576ca860c446e1d24d9bc277fa9b9e661899cb49affd9 /exec failed with exit code 2

CANCELLED JOB - The following job was cancelled. Please retry:

• periodic (gh)

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

[meta-codesync]

@sxu has exported this pull request. If you are a Meta employee, you can view the originating Diff in D84211619.