minor fix on index hoisting

csarofeen · Sep 21, 2022 · 4be509b · 4be509b
1 parent ed8b2e4
commit 4be509b
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/torch/csrc/jit/codegen/cuda/lower_mem_index.cpp b/torch/csrc/jit/codegen/cuda/lower_mem_index.cpp
@@ -785,11 +785,6 @@ bool isSeparableSmemSwizzledProducerIndex(
     auto exact_producer_it = exact_to_producer_id_map.find(
         ir_utils::caMapExactConcreteId(split->in()));
 
-    if (exact_producer_it == exact_to_producer_id_map.end()) {
-      // Returns early if no exact mapped id is found.
-      return false;
-    }
-
     // Find a tile split node that this needs to assume.
     while (exact_producer_it == exact_to_producer_id_map.end()) {
       auto in_def = split->in()->definition();