Skip to content

Commit ab3c57f

Browse files
njriasanfacebook-github-bot
authored andcommitted
Update matmul_fp8_row_meta signature (#4168)
Summary: Pull Request resolved: #4168 X-link: facebookresearch/FBGEMM#1248 Updates the signature to matmul_fp8_row_meta to be consistent with matmul_fp8_row. Differential Revision: D75030969
1 parent b9a2985 commit ab3c57f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

fbgemm_gpu/experimental/gemm/triton_gemm/fp8_gemm.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1557,7 +1557,9 @@ def matmul_fp8_row_meta(
15571557
allow_tf32: bool = True,
15581558
fp8_fast_accum: bool = True,
15591559
imprecise_acc: bool = False,
1560-
tma_persistent: bool = False,
1560+
tma_persistent: bool = True,
1561+
no_use_persistent: Optional[bool] = None,
1562+
use_warp_specialization: bool = False,
15611563
) -> torch.Tensor:
15621564
"""Shape function for torch compile."""
15631565
M, K = a.shape

0 commit comments

Comments
 (0)