Eliminate use of deprecated 'device' argument in Tensor.pin_memory()

[drivanov]

A crash was observed due to the usage of the deprecated device parameter.

../opt/pyg/pytorch_geometric/test/test_edge_index.py::test_data_loader[True-0-int64] DeprecationWarning: The argument 'device' of Tensor.pin_memory() is deprecated. Please do not pass this argument. (Triggered internally at /opt/pytorch/pytorch/aten/src/ATen/native/Memory.cpp:46.)
DeprecationWarning: The argument 'device' of Tensor.is_pinned() is deprecated. Please do not pass this argument. (Triggered internally at /opt/pytorch/pytorch/aten/src/ATen/native/Memory.cpp:31.)
Traceback (most recent call last):
  File "/usr/local/lib/python3.12/dist-packages/torch/utils/data/_utils/pin_memory.py", line 91, in pin_memory
    clone[i] = pin_memory(item, device)
               ^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.12/dist-packages/torch/utils/data/_utils/pin_memory.py", line 57, in pin_memory
    return data.pin_memory(device)
           ^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.12/dist-packages/torch_geometric/edge_index.py", line 1211, in __torch_dispatch__
    return HANDLED_FUNCTIONS[func](*args, **(kwargs or {}))
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: _pin_memory() takes 1 positional argument but 2 were given

This PR removes its use to ensure compatibility with current PyTorch versions

[pytorch-bot]

🔗 Helpful Links

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

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

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

✅ No Failures

As of commit be83b24 with merge base 86db4de ():
💚 Looks good so far! There are no failures yet. 💚

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

[linux-foundation-easycla]

The committers listed above are authorized under a signed CLA.

• ✅ login: drivanov / name: Andrei Ivanov (25a28b5, 1e65d77, 7baab64, be83b24, fc11faf)

[eqy]

Would it be possible to add a test for this?

[drivanov]

Would it be possible to add a test for this?

6 tests added

[eqy]

Does it make sense to just add the tests to test_dataloader.py rather than creating a new file? https://github.com/pytorch/pytorch/blob/ebddbe787a1d193a56b3ed60986726b7a4b61ba3/test/test_dataloader.py

[drivanov]

Does it make sense to just add the tests to test_dataloader.py rather than creating a new file? https://github.com/pytorch/pytorch/blob/ebddbe787a1d193a56b3ed60986726b7a4b61ba3/test/test_dataloader.py

I can make that change if you prefer, although I noticed that test_dataloader.py is already quite large (≈approximately 3680 lines).

[drivanov]

@eqy : Could you, please, review this PR?

[divyanshk]

@drivanov Seconding @eqy here. lets add it to test_dataloader.py it already has other unit tests on pin memory. We can regroup tests here later.

Also, might have to check if we should update the docstring as well

[drivanov]

@eqy : I moved the 6 pin_memory tests to test_dataloader.py

[ngimel]

For the future, AI assistance on PRs has to be disclosed