tiny fix

tenstorrent · Dec 25, 2024 · 04c953a · 04c953a
1 parent ce9cdac
commit 04c953a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/lib/Conversion/TTIRToTTNN/TTIRToTTNN.cpp b/lib/Conversion/TTIRToTTNN/TTIRToTTNN.cpp
@@ -75,10 +75,9 @@ class TensorEmptyConversionPattern
     ttnn::ShardSpecAttr shardSpecAttr = ttnn::ShardSpecAttr::get(
         op.getContext(),
         ttnn::ShapeAttr::get(op.getContext(), layoutAttr.getShardShape()));
-    ttnn::TensorMemoryLayoutAttr memLayout =
-        layoutAttr.getMemLayout() ? layoutAttr.getMemLayout() : nullptr;
-    ttnn::MemoryConfigAttr memoryConfigAttr = ttnn::MemoryConfigAttr::get(
-        op.getContext(), bufferTypeAttr, shardSpecAttr, memLayout);
+    ttnn::MemoryConfigAttr memoryConfigAttr =
+        ttnn::MemoryConfigAttr::get(op.getContext(), bufferTypeAttr,
+                                    shardSpecAttr, layoutAttr.getMemLayout());
 
     // Replace op
     //