Skip to content

Commit e2960d5

Browse files
minor linting
Signed-off-by: cliu-us <[email protected]>
1 parent 434e68f commit e2960d5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fms_mo/custom_ext_kernels/triton_kernels.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -499,10 +499,10 @@ def isPowerofTwo(x):
499499
mm_kernel = imatmul_kernel
500500
else:
501501
acc_dtype = torch.float32
502-
mm_kernel = matmul_kernel if c == None else matmul_kernel_DABC
502+
mm_kernel = matmul_kernel if c is None else matmul_kernel_DABC
503503
assert chunk_trun_bits < 23, "FP32 accumulator only has 23 mantissa bits"
504504

505-
if c == None:
505+
if c is None:
506506
c_org_dtype = a.dtype
507507
c = torch.zeros((M, N), device=a.device, dtype=acc_dtype)
508508
else:

0 commit comments

Comments
 (0)