Commit 0d56855b authored by sangwzh's avatar sangwzh
Browse files

update indices' ptr getting

parent 52639ff6
...@@ -160,7 +160,7 @@ std::tuple<torch::Tensor, torch::Tensor> UVAIndexSelectCSCCopyIndices( ...@@ -160,7 +160,7 @@ std::tuple<torch::Tensor, torch::Tensor> UVAIndexSelectCSCCopyIndices(
_CopyIndicesAlignedKernel, grid, block, 0, _CopyIndicesAlignedKernel, grid, block, 0,
static_cast<indptr_t>(edge_count_aligned_), sliced_indptr, static_cast<indptr_t>(edge_count_aligned_), sliced_indptr,
output_indptr.data_ptr<indptr_t>(), output_indptr_aligned_ptr, output_indptr.data_ptr<indptr_t>(), output_indptr_aligned_ptr,
reinterpret_cast<indices_t*>(cuda::getTensorDevicePointer<indices_t>(indices)), reinterpret_cast<indices_t*>(indices.data_ptr()),
coo_aligned_rows.data_ptr<coo_rows_t>(), coo_aligned_rows.data_ptr<coo_rows_t>(),
reinterpret_cast<indices_t*>(output_indices.data_ptr()), perm); reinterpret_cast<indices_t*>(output_indices.data_ptr()), perm);
})); }));
......
...@@ -325,7 +325,7 @@ c10::intrusive_ptr<sampling::FusedSampledSubgraph> SampleNeighbors( ...@@ -325,7 +325,7 @@ c10::intrusive_ptr<sampling::FusedSampledSubgraph> SampleNeighbors(
.data_ptr<probs_t>(); .data_ptr<probs_t>();
} }
const indices_t* indices_ptr = const indices_t* indices_ptr =
layer ? cuda::getTensorDevicePointer<indices_t>(indices) : nullptr; layer ? indices.data_ptr<indices_t>() : nullptr;
const dim3 block(BLOCK_SIZE); const dim3 block(BLOCK_SIZE);
const dim3 grid( const dim3 grid(
(num_edges.value() + BLOCK_SIZE - 1) / (num_edges.value() + BLOCK_SIZE - 1) /
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment