diff --git a/tests/cpp/test_outer_reduction.cpp b/tests/cpp/test_outer_reduction.cpp index f4ee96b0150..5c63556bde2 100644 --- a/tests/cpp/test_outer_reduction.cpp +++ b/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; diff --git a/tests/cpp/test_persistent_buffer.cpp b/tests/cpp/test_persistent_buffer.cpp index ebfff71a32a..1f92003e9b6 100644 --- a/tests/cpp/test_persistent_buffer.cpp +++ b/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; diff --git a/tests/cpp/test_reduction.cpp b/tests/cpp/test_reduction.cpp index 005710474f5..0adaf9815a0 100644 --- a/tests/cpp/test_reduction.cpp +++ b/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; diff --git a/tests/cpp/test_reduction_pointwise.cpp b/tests/cpp/test_reduction_pointwise.cpp index 1f98d7ef9ca..3cac57a0ab4 100644 --- a/tests/cpp/test_reduction_pointwise.cpp +++ b/tests/cpp/test_reduction_pointwise.cpp @@ -16,7 +16,13 @@ #include 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 diff --git a/tests/cpp/test_serial_gridreduce.cpp b/tests/cpp/test_serial_gridreduce.cpp index 13ee5d77df9..1d1291c46d2 100644 --- a/tests/cpp/test_serial_gridreduce.cpp +++ b/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}) { diff --git a/tests/cpp/test_welford.cpp b/tests/cpp/test_welford.cpp index 9f13383fc62..07daa7b6775 100644 --- a/tests/cpp/test_welford.cpp +++ b/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;