fix replace errors x3

tenstorrent · Mar 6, 2025 · 2a323dd · 2a323dd
1 parent 539dd14
commit 2a323dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/demos/t3000/falcon40b/tests/test_falcon_decoder.py b/models/demos/t3000/falcon40b/tests/test_falcon_decoder.py
@@ -216,7 +216,7 @@ def run_test_FalconDecoder_inference(
             layout=ttnn.TILE_LAYOUT,
             device=mesh_device,
             memory_config=model_config["KV_CACHE_MEMCFG"],
-            mesh_mapper=(mesh_device, dim=1),
+            mesh_mapper=ttnn.shard_tensor_to_mesh_mapper(mesh_device, dim=1),
         )
 
         tt_layer_past = (tt_k_cache, tt_v_cache)