Short name describing what triggered the graph break
flex_gemm: improper args/kwargs
Values or code snippet captured at the break point
args: {args}, kwargs: {kwargs}
Explanation of why the graph break was triggered
flex_gemm expects 5 positional arguments: gemm_op, body_fn, args, kwargs, kernel_options. args is expected to be list/tuple and kwargs/kernel_options are expected to be dicts.
Hints on how to resolve the graph break
torch.compile call, or by using torch.compiler.set_stance("force_eager").