Skip to content

GitLab

Unverified Commit f281959a authored by Ramon Zhou's avatar Ramon Zhou Committed by GitHub
Browse files

[Graphbolt] Delete the expensive check (#6264)

parent 240e28a2
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
graphbolt/src/csc_sampling_graph.cc
View file @ f281959a
...@@ -403,12 +403,6 @@ c10::intrusive_ptr<SampledSubgraph> CSCSamplingGraph::SampleNeighbors( ...@@ -403,12 +403,6 @@ c10::intrusive_ptr<SampledSubgraph> CSCSamplingGraph::SampleNeighbors(
probs_or_mask.value().dtype() == torch::kFloat16) { probs_or_mask.value().dtype() == torch::kFloat16) {
probs_or_mask = probs_or_mask.value().to(torch::kFloat32); probs_or_mask = probs_or_mask.value().to(torch::kFloat32);
} }
TORCH_CHECK(
((probs_or_mask.value().max() < INFINITY) &
(probs_or_mask.value().min() >= 0))
.item()
.to<bool>(),
"Invalid probs_or_mask (contains either `inf`, `nan` or element < 0).");
} }
if (layer) { if (layer) {
......
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