Minor bugfix in transform_rfactor.cpp (#1715)

`found_non_rfactor_reduction` is used to detect errors when all reduction dims are marked as rfactors. However, this code is not finding non-rfactor reduction, but instead arbitrary reduction. Fortunately, other parts of our code could detect the same error, so this bug does not have any real effect. But still, I think we need to fix this.

torch/csrc/jit/codegen/cuda/transform_rfactor.cpp

@@ -196,10 +196,11 @@ TensorDomain* TransformRFactor::runReplay(
   {
     size_t i = 0;
     for (auto id : orig_td->domain()) {
-      if (axes_set.find(i++) != axes_set.end())
+      if (axes_set.find(i++) != axes_set.end()) {
         rfactor_axes.emplace(id);
-      if (id->isReduction())
+      } else if (id->isReduction()) {
         found_non_rfactor_reduction = true;
+      }
     }
   }