Skip to content

Commit 9ee258b

Browse files
Merge pull request #166 from chichun-charlie-liu/main
fix: pylint false alarm on libdevice functions
2 parents 78214ae + f48a948 commit 9ee258b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fms_mo/custom_ext_kernels/triton_kernels.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -449,7 +449,7 @@ def round_and_trun_mask(chunk_trun_bits, clamp_acc_to_dl16):
449449
@triton.jit
450450
def round_and_trun(x, round_bit, trun_mask, clamp_acc_to_dl16):
451451
"""Round and truncate (usually for accumulator)."""
452-
x = libdevice.uint_as_float((libdevice.float_as_uint(x) + round_bit) & trun_mask)
452+
x = libdevice.uint_as_float((libdevice.float_as_uint(x) + round_bit) & trun_mask) # pylint: disable=assignment-from-no-return
453453

454454
if clamp_acc_to_dl16:
455455
# clamp to DL16 min/max:

0 commit comments

Comments
 (0)