// Copyright 2026 The IREE Authors
//
// Licensed under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#ifndef IREE_COMPILER_CODEGEN_DIALECT_MAP_TRANSFORMS_MAPPATTERNS_H_
#define IREE_COMPILER_CODEGEN_DIALECT_MAP_TRANSFORMS_MAPPATTERNS_H_

#include "mlir/IR/PatternMatch.h"
#include "mlir/IR/Value.h"

namespace mlir::iree_compiler {

/// Adds PackLayout-specific patterns for distributing generic vector ops
/// (shape_cast, broadcast, transpose, step) using PackLayoutAttr.
void populateMapDistributeGenericPatterns(RewritePatternSet &patterns,
                                          Value threadId);

} // namespace mlir::iree_compiler
#endif // IREE_COMPILER_CODEGEN_DIALECT_MAP_TRANSFORMS_MAPPATTERNS_H_
