Skip to content

Commit 678885a

Browse files
committed
Specify mesh_type when opening t3k device, update tt-metal commit
Signed-off-by: Salar Hosseini <skhorasgani@tenstorrent.com>
1 parent 5dd58b9 commit 678885a

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

tt_metal/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
## vLLM and tt-metal Branches
33
Git-checkout the following branches in each repo separately:
44
- vLLM branch: [dev](https://github.com/tenstorrent/vllm/tree/dev) (last verified commit: [3f7beb2](https://github.com/tenstorrent/vllm/tree/3f7beb23cbaf3be2e104061905da5f91644e5a68))
5-
- tt-metal branch: [main](https://github.com/tenstorrent/tt-metal) (last verified commit: [0107040](https://github.com/tenstorrent/tt-metal/tree/01070409e582616d8962f371e8497abbf252bb81))
5+
- tt-metal branch: [main](https://github.com/tenstorrent/tt-metal) (last verified commit: [f521af0](https://github.com/tenstorrent/tt-metal/tree/f521af0061bf53567942b7a27fd89aa300ec16ce))
66

77
## Environment Creation
88

vllm/worker/tt_worker.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -376,6 +376,7 @@ def _open_t3k_mesh_device(self):
376376
ttnn.MeshShape(2, 4),
377377
dispatch_core_type=self._get_dispatch_core_type(),
378378
**device_params,
379+
mesh_type=ttnn.MeshType.Ring,
379380
)
380381
logger.debug(f"multidevice with {mesh_device.get_num_devices()} devices is created")
381382
return mesh_device

0 commit comments

Comments
 (0)