From 30aebbce82a903c97be9d0a5814766a85300c31d Mon Sep 17 00:00:00 2001 From: "pengwa@microsoft.com" Date: Mon, 4 Dec 2023 09:46:42 +0000 Subject: [PATCH] enable tile recompute UT --- .../orttraining/test/optimizer/memory_optimizer_test.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/orttraining/orttraining/test/optimizer/memory_optimizer_test.cc b/orttraining/orttraining/test/optimizer/memory_optimizer_test.cc index 82e9b40d19586..d5f0ebc023116 100644 --- 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(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;