Add mechanism for overriding with corresponding unittests.

tenstorrent · Nov 26, 2024 · 124f0e9 · 124f0e9
1 parent 742f809
commit 124f0e9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/unittests/TestOptimizerOverrides/TestOptimizerOverrides.cpp b/test/unittests/TestOptimizerOverrides/TestOptimizerOverrides.cpp
@@ -11,10 +11,10 @@ using namespace mlir::tt::ttnn;
 class TestOptimizerOverrides : public ::testing::Test {
 
 public:
-  OptimizerOverridesHandler *optimizerOverridesHandler;
+  std::shared_ptr<OptimizerOverridesHandler> optimizerOverridesHandler;
 
   void SetUp() override {
-    optimizerOverridesHandler = new OptimizerOverridesHandler();
+    optimizerOverridesHandler = std::make_shared<OptimizerOverridesHandler>();
   }
 
   llvm::StringMap<InputLayoutOverrideParams> createInputLayoutOverrides() {
@@ -194,7 +194,7 @@ class TestOptimizerOverrides : public ::testing::Test {
     return true;
   }
 
-  void TearDown() override { delete optimizerOverridesHandler; }
+  void TearDown() override {}
 };
 
 // Test the setOptimizerPass method