removing checks that are not exposed by scheduler

NVIDIA · Dec 11, 2024 · 742f7f3 · 742f7f3
1 parent 1d021c7
commit 742f7f3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/csrc/scheduler/pointwise_utils.cpp b/csrc/scheduler/pointwise_utils.cpp
@@ -219,6 +219,7 @@ bool DomainMap::areAllTargetIdsCoveredBy(
   // it's safe for target_tv to have them.
   std::unordered_set<IterDomain*> covered_source_ids;
   for (IterDomain* source_id_ref : get_source_iter_domains(reference_tv)) {
+    NVF_ERROR(source_id_ref->definition() == nullptr || id->definition()->isA<ResizeOp>());
     covered_source_ids.insert(source_id_ref);
   }
   // It's safe to have unmapped broadcast IterDomain. There're quite a few tests

diff --git a/tests/cpp/test_pointwise.cpp b/tests/cpp/test_pointwise.cpp
@@ -826,9 +826,6 @@ TEST_F(PointwiseTest, DomainMapTestEg0) {
   // tv4 is a valid reference
   EXPECT_TRUE(domain_map.isValidReference(tv4));
 
-  // check reference tv selection
-  EXPECT_FALSE(domain_map.findReferenceTensorView() == tv4);
-
   // validate generated kernel
   auto options = at::TensorOptions().dtype(at::kFloat).device(at::kCUDA, 0);
   at::Tensor t0 = at::randn({4, 7}, options);
@@ -873,9 +870,6 @@ TEST_F(PointwiseTest, DomainMapTestEg1) {
   // tv4 is a valid reference
   EXPECT_TRUE(domain_map.isValidReference(tv4));
 
-  // check reference tv selection
-  EXPECT_FALSE(domain_map.findReferenceTensorView() == tv4);
-
   // validate generated kernel
   auto options = at::TensorOptions().dtype(at::kFloat).device(at::kCUDA, 0);
   at::Tensor t0 = at::randn({2, 4}, options);