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

[janeyx99]

I KNOW THIS DOES NOT BUILD

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.

❌ 6 New Failures

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

NEW FAILURES - The following jobs have failed:

• PR Label Check / Check PR Labels (gh)
  Process completed with exit code 1.
• 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 b221deeaff14873489e0200b8efe58e076fbb986c47a7053d8b9f1592df537fe /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 ef11ca9cf18462136897de545f8f0d0acc25063d9b842d135910bebaa78533fa /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 d680a855bc4ced494e1a1b0c6ebb279bb97fede1a5b619bd57be1d645d8db827 /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 f672d6bc361fe666a3fa8ef29273a0e323eef81e9baabf8b4598c64da0304da3 /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 b16a96c9bb01358db4581fb97b6ef55b5900f764f1de9427e013e28a8b76b3cc /exec failed with exit code 1

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