enable tile recompute UT

microsoft · Dec 4, 2023 · 30aebbc · 30aebbc
1 parent 7c3bbc3
commit 30aebbc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/orttraining/orttraining/test/optimizer/memory_optimizer_test.cc b/orttraining/orttraining/test/optimizer/memory_optimizer_test.cc
@@ -92,8 +92,7 @@ TEST(MemoryOptimizerTests, GeluRecompute) {
   ASSERT_EQ(original_gelu_node->Priority(), static_cast<int>(ExecutionPriority::DEFAULT));
 }
 
-// Disable this UT for now. It has strong dependency on graph topological order, which is not correct logically.
-TEST(MemoryOptimizerTests, DISABLED_TileRecompute) {
+TEST(MemoryOptimizerTests, TileRecompute) {
   const logging::Logger* logger = &logging::LoggingManager::DefaultLogger();
   auto model_uri = MODEL_FOLDER "recompute_tile.onnx";
   std::shared_ptr<Model> model;