[super ugly maybe working code] use shim.h instead of Tensor

[janeyx99]

I KNOW THIS DOES NOT BUILD -- this should build now!

this is a poc change to see what making custom ops use aoti shim.h would look like

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1548

• 📄 Preview Python docs built from this PR

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

❌ 5 New Failures

As of commit e1c39be with merge base 1beb6d0 ():

NEW FAILURES - The following jobs have failed:

• Run Float8 Tests / test (SM-89, linux.g6.4xlarge.experimental.nvidia.gpu, --pre torch --index-url https://download.p... / linux-job (gh)
  RuntimeError: Command docker exec -t 94c980f258ba00043eedac83c375989117bffbce1012e4b8b4ae52597a896b78 /exec failed with exit code 1
• Run Regression Tests / test (CUDA 2.3, linux.g5.12xlarge.nvidia.gpu, torch==2.3.0, cuda, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t e5ddffceb911fda1afb14a3a8cf1a4fbebeb5991953328ccdd74f7079815e4d7 /exec failed with exit code 1
• Run Regression Tests / test (CUDA 2.4, linux.g5.12xlarge.nvidia.gpu, torch==2.4.0, cuda, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 0363bc14a1dafb420aeb3a4245c06e888206a4958cd56e031124168a21252426 /exec failed with exit code 1
• Run Regression Tests / test (CUDA 2.5.1, linux.g5.12xlarge.nvidia.gpu, torch==2.5.1 --index-url https://download.pytorch... / linux-job (gh)
  RuntimeError: Command docker exec -t d04aca3d335efa4ac706f4fe50e01dff99852d5374a2b55a3acb53673837249f /exec failed with exit code 1
• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://downloa... / linux-job (gh)
  RuntimeError: Command docker exec -t f19515e084535ba40087d728cc9813e5c41ecf3a2605a3b7b09c230cf1d1fce7 /exec failed with exit code 1

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

[janeyx99]

UGH IGNORE the random VSCode formats--Idk how autoformat is turned on in this repo for me + will figure it out later, skip this line til 170!!!