Skip to content

Commit 9326f49

Browse files
committed
Apply code-format changes
1 parent 068db00 commit 9326f49

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

python/test/tle/unit/test_tle_get_node_rank.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,15 +6,12 @@
66
import triton.experimental.tle.language as tle
77
import triton.language as tl
88

9-
109
LOCAL_WORLD_SIZE = int(os.environ["LOCAL_WORLD_SIZE"])
1110
WORLD_SIZE = int(os.environ["WORLD_SIZE"])
1211
if WORLD_SIZE % LOCAL_WORLD_SIZE != 0:
1312
raise ValueError("WORLD_SIZE must be divisible by LOCAL_WORLD_SIZE")
1413

15-
DEVICE_MESH = tle.device_mesh(
16-
tle.MeshConfig(node=WORLD_SIZE // LOCAL_WORLD_SIZE, device=LOCAL_WORLD_SIZE)
17-
)
14+
DEVICE_MESH = tle.device_mesh(tle.MeshConfig(node=WORLD_SIZE // LOCAL_WORLD_SIZE, device=LOCAL_WORLD_SIZE))
1815

1916

2017
@triton.jit

0 commit comments

Comments
 (0)