def METHOD_NAME(context: Context, p_output: StorePartition, n: int) -> None:
    launch_domain = Rect((n, n))
    task = context.create_manual_task(
        CuNumericOpCode.TRILU, launch_domain=launch_domain
    )
    task.add_output(p_output)
    task.add_input(p_output)
    task.add_scalar_arg(True, ty.bool_)
    task.add_scalar_arg(0, ty.int32)
    # Add a fake task argument to indicate that this is for Cholesky
    task.add_scalar_arg(True, ty.bool_)
    task.execute()