Skip to content

Commit a4ea093

Browse files
bug fix
Signed-off-by: cliu-us <[email protected]>
1 parent eea8426 commit a4ea093

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

fms_mo/custom_ext_kernels/triton_kernels.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -420,6 +420,7 @@ def round_and_trun(x, round_bit, trun_mask):
420420
def tl_matmul_chunk_truncate(
421421
a,
422422
b,
423+
c=None,
423424
activation="",
424425
chunk_trun_bits=0,
425426
chunk_size=16,

0 commit comments

Comments
 (0)