[Inductor UT] Generalize device-bias code in test_triton_kernels.py and test_torchinductor.py

[etaf]

Stack from ghstack (oldest at bottom):

• [Inductor] Handle device_put op in constant folding. #130824
• -> [Inductor UT] Generalize device-bias code in test_triton_kernels.py and test_torchinductor.py #130817

[Inductor UT] Generalize newly introduced device-bias code in test_triton_kernels.py::test_add_kernel and test_torchinductor.py::test_ctr_not_moved_to_cuda_when_used_in_index_put
Fix #130814 , #130838

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

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

❌ 3 New Failures, 1 Unrelated Failure

As of commit 99f9396 with merge base e11c410 ():

NEW FAILURES - The following jobs have failed:

• xpu / linux-jammy-xpu-py3.8 / test (default, 1, 4, linux.idc.xpu) (gh)
  inductor/test_torchinductor.py::GPUTests::test_uint_xpu
• xpu / linux-jammy-xpu-py3.8 / test (default, 3, 4, linux.idc.xpu) (gh)
  inductor/test_torchinductor_dynamic_shapes.py::DynamicShapesGPUTests::test_uint_dynamic_shapes_xpu
• xpu / linux-jammy-xpu-py3.8 / test (default, 4, 4, linux.idc.xpu) (gh)
  inductor/test_torchinductor_codegen_dynamic_shapes.py::DynamicShapesCodegenGPUTests::test_uint_dynamic_shapes_xpu

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

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

• Lint / lintrunner-noclang / linux-job (gh) (trunk failure)
  >>> Lint for torch/_functorch/_aot_autograd/subclass_utils.py:

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

[etaf]

The three failed job:
xpu / linux-jammy-xpu-py3.8 / test (default, 1, 4, linux.idc.xpu) (gh)
inductor/test_torchinductor.py::GPUTests::test_uint_xpu
xpu / linux-jammy-xpu-py3.8 / test (default, 3, 4, linux.idc.xpu) (gh)
inductor/test_torchinductor_dynamic_shapes.py::DynamicShapesGPUTests::test_uint_dynamic_shapes_xpu
xpu / linux-jammy-xpu-py3.8 / test (default, 4, 4, linux.idc.xpu) (gh)
inductor/test_torchinductor_codegen_dynamic_shapes.py::DynamicShapesCodegenGPUTests::test_uint_dynamic_shapes_xpu

has been fixed in the next stacked PR. Please ignore them.