Skip to content

Commit

Permalink
[FIX] Remove Unnecessary CUDA Check Causing Deprecation Warning (rapi…
Browse files Browse the repository at this point in the history
…dsai#4239)

There is an unnecessary cuda check made in the cuGraph-PyG `CuGraphStore` object causing a deprecation warning.  This PR removes the check.

Authors:
  - Alex Barghi (https://github.com/alexbarghi-nv)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)

URL: rapidsai#4239
  • Loading branch information
alexbarghi-nv authored Apr 14, 2024
1 parent 3e9e7ac commit 2578421
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,9 +336,7 @@ def __del__(self):
def __make_offsets(self, input_dict):
offsets = {}
offsets["stop"] = [input_dict[v] for v in sorted(input_dict.keys())]
offsets["stop"] = torch.tensor(offsets["stop"])
if torch.has_cuda:
offsets["stop"] = offsets["stop"].cuda()
offsets["stop"] = torch.tensor(offsets["stop"]).cuda()

cumsum = offsets["stop"].cumsum(0)
offsets["start"] = cumsum - offsets["stop"]
Expand Down

0 comments on commit 2578421

Please sign in to comment.