Allow GPU input in NodeLoader and LinkLoader

[rusty1s]

Fixes #7557

[codecov]

Codecov Report

Merging #7572 (5573540) into master (0549077) will decrease coverage by 0.33%.
The diff coverage is 86.66%.

@@            Coverage Diff             @@
##           master    #7572      +/-   ##
==========================================
- Coverage   91.73%   91.41%   -0.33%     
==========================================
  Files         449      449              
  Lines       25097    25119      +22     
==========================================
- Hits        23023    22962      -61     
- Misses       2074     2157      +83     

Impacted Files	Coverage Δ
torch_geometric/config_store.py	90.06% <ø> (ø)
torch_geometric/graphgym/__init__.py	100.00% <ø> (ø)
torch_geometric/loader/link_loader.py	98.01% <ø> (ø)
torch_geometric/profile/benchmark.py	78.46% <0.00%> (-1.23%)	⬇️
torch_geometric/sampler/base.py	98.20% <96.29%> (-0.28%)	⬇️

... and 16 files with indirect coverage changes

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more