[Graphbolt] Delete the expensive check (#6264)

f281959a · Ramon Zhou · GitHub · 240e28a2 · f281959a
Unverified Commit f281959a authored Sep 01, 2023 by Ramon Zhou Committed by GitHub Sep 01, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 6 deletions

graphbolt/src/csc_sampling_graph.cc graphbolt/src/csc_sampling_graph.cc +0 -6

No files found.
--- a/graphbolt/src/csc_sampling_graph.cc
+++ b/graphbolt/src/csc_sampling_graph.cc
@@ -403,12 +403,6 @@ c10::intrusive_ptr<SampledSubgraph> CSCSamplingGraph::SampleNeighbors(
        probs_or_mask.value().dtype() == torch::kFloat16) {
      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) {