def extract_config(func_attrs, dtype="float16"):
    """Extracts cutlass config for conv kernels."""
    import cutlass_lib

    return conv2d_extract_config(
        func_attrs=func_attrs,
        dtype=dtype,
        op_kind=cutlass_lib.library.OperationKind.Conv3d,
        op_layout="ndhwc",
    )
Extracts cutlass config for conv kernels.
python/aitemplate/backend/cuda/conv3d/common.py