Enable TensorIndexer with some of the reduction-related tests (#5573)

Confirmed code diff results are benign

Author: naoyam

tests/cpp/test_outer_reduction.cpp

@@ -32,7 +32,13 @@
 
 namespace nvfuser {
 
-using OuterReductionTest = NVFuserTest;
+class OuterReductionTest : public NVFuserTest {
+ protected:
+  void SetUp() override {
+    NVFuserTest::SetUp();
+    EnableOptionsGuard::getCurOptions().set(EnableOption::IdModel, {"all"});
+  }
+};
 
 using namespace at::indexing;
 

tests/cpp/test_persistent_buffer.cpp

@@ -23,7 +23,14 @@ namespace nvfuser {
 
 using testing::Contains;
 using testing::UnorderedElementsAre;
-using PersistentBufferTest = NVFuserTest;
+
+class PersistentBufferTest : public NVFuserTest {
+ protected:
+  void SetUp() override {
+    NVFuserTest::SetUp();
+    EnableOptionsGuard::getCurOptions().set(EnableOption::IdModel, {"all"});
+  }
+};
 
 TEST_F(PersistentBufferTest, FusionPersistentBufferCalculation1_CUDA) {
   Fusion fusion;

tests/cpp/test_reduction.cpp

@@ -73,7 +73,13 @@ void validateNoParallelBroadcastExist(kir::Kernel* kernel) {
 
 } // namespace
 
-using ReductionTest = NVFuserTest;
+class ReductionTest : public NVFuserTest {
+ protected:
+  void SetUp() override {
+    NVFuserTest::SetUp();
+    EnableOptionsGuard::getCurOptions().set(EnableOption::IdModel, {"all"});
+  }
+};
 
 TEST_F(ReductionTest, GridAllreduce1) {
   const int nx = 999;

tests/cpp/test_reduction_pointwise.cpp

@@ -16,7 +16,13 @@
 #include <tests/cpp/validator.h>
 namespace nvfuser {
 
-using PointwiseFusedReductionTest = NVFuserTest;
+class PointwiseFusedReductionTest : public NVFuserTest {
+ protected:
+  void SetUp() override {
+    NVFuserTest::SetUp();
+    EnableOptionsGuard::getCurOptions().set(EnableOption::IdModel, {"all"});
+  }
+};
 
 // inner reduction + non-broadcast epilogue, can't be fused
 // outer reduction + non-broadcast epilogue, can be fused

tests/cpp/test_serial_gridreduce.cpp

@@ -33,7 +33,13 @@
 
 namespace nvfuser {
 
-using SerialGridReductionTest = NVFuserTest;
+class SerialGridReductionTest : public NVFuserTest {
+ protected:
+  void SetUp() override {
+    NVFuserTest::SetUp();
+    EnableOptionsGuard::getCurOptions().set(EnableOption::IdModel, {"all"});
+  }
+};
 
 TEST_F(SerialGridReductionTest, Scheduling) {
   for (bool serial : {true, false}) {

tests/cpp/test_welford.cpp

@@ -15,7 +15,13 @@
 
 namespace nvfuser {
 
-using WelfordTest = NVFuserTest;
+class WelfordTest : public NVFuserTest {
+ protected:
+  void SetUp() override {
+    NVFuserTest::SetUp();
+    EnableOptionsGuard::getCurOptions().set(EnableOption::IdModel, {"all"});
+  }
+};
 
 TEST_F(WelfordTest, SerialWelford) {
   int x = 128, y = 64, z = 64;