Skip to content

Commit 9b6f23f

Browse files
Apply Ruff auto-fixes
1 parent 5eb6336 commit 9b6f23f

File tree

5 files changed

+9
-9
lines changed

5 files changed

+9
-9
lines changed

iris/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,4 +106,4 @@
106106
"INFO",
107107
"WARNING",
108108
"ERROR",
109-
]
109+
]

iris/iris.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1555,22 +1555,22 @@ def copy(src_ptr, dst_ptr, from_rank, to_rank, cur_rank, heap_bases, mask=None):
15551555

15561556
trap_if((cur_rank != from_rank) and (cur_rank != to_rank))
15571557

1558-
cur_base = tl.load(heap_bases + cur_rank)
1558+
cur_base = tl.load(heap_bases + cur_rank)
15591559

15601560
from_base = tl.load(heap_bases + from_rank)
1561-
to_base = tl.load(heap_bases + to_rank)
1561+
to_base = tl.load(heap_bases + to_rank)
15621562

15631563
src_ptr_int = tl.cast(src_ptr, tl.uint64)
15641564
src_offset = src_ptr_int - cur_base
15651565

15661566
dst_ptr_int = tl.cast(dst_ptr, tl.uint64)
1567-
dst_offset = dst_ptr_int - cur_base
1567+
dst_offset = dst_ptr_int - cur_base
15681568

15691569
from_base_byte = tl.cast(from_base, tl.pointer_type(tl.int8))
1570-
to_base_byte = tl.cast(to_base , tl.pointer_type(tl.int8))
1570+
to_base_byte = tl.cast(to_base, tl.pointer_type(tl.int8))
15711571

15721572
translated_src = tl.cast(from_base_byte + src_offset, src_ptr.dtype)
1573-
translated_dst = tl.cast(to_base_byte + dst_offset, src_ptr.dtype)
1573+
translated_dst = tl.cast(to_base_byte + dst_offset, src_ptr.dtype)
15741574

15751575
data = tl.load(translated_src, mask=mask)
15761576
tl.store(translated_dst, data, mask=mask)

tests/unittests/test_copy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def copy_kernel(
2525
for target_rank in range(num_ranks):
2626
src_data = data + BLOCK_SIZE * cur_rank
2727
dest_data = results + BLOCK_SIZE * target_rank
28-
iris.copy(src_data + offsets, dest_data + offsets, target_rank, cur_rank, cur_rank, heap_bases, mask)
28+
iris.copy(src_data + offsets, dest_data + offsets, target_rank, cur_rank, cur_rank, heap_bases, mask)
2929

3030

3131
@pytest.mark.parametrize(

tests/unittests/test_get.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,4 +81,4 @@ def test_get_api(dtype, BLOCK_SIZE):
8181
print(e)
8282
print("Expected:", expected)
8383
print("Actual:", results)
84-
raise
84+
raise

tests/unittests/test_put.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,4 +75,4 @@ def test_put_api(dtype, BLOCK_SIZE):
7575
print(e)
7676
print("Expected:", expected)
7777
print("Actual:", results)
78-
raise
78+
raise

0 commit comments

Comments
 (0)