e259175995
This patch fixes a crash in the SCF to OpenMP conversion pass when encountering scf.parallel with vector reductions. - Extracts scalar element types for bitwidth calculations. - Uses DenseElementsAttr for vector splat initializers. - Bypasses llvm.atomicrmw for vector types (not supported in LLVM IR). Fixes #173860 --------- Co-authored-by: Aniket Singh <amiket.singh.3200.00@gmail.com>
29 lines
1.0 KiB
MLIR
29 lines
1.0 KiB
MLIR
// RUN: mlir-opt %s --convert-scf-to-openmp | FileCheck %s
|
|
|
|
// CHECK-LABEL: omp.declare_reduction @__scf_reduction : vector<2xi1>
|
|
// CHECK: init {
|
|
// CHECK: %[[INIT:.*]] = llvm.mlir.constant(dense<true> : vector<2xi1>) : vector<2xi1>
|
|
// CHECK: omp.yield(%[[INIT]] : vector<2xi1>)
|
|
// CHECK: }
|
|
// CHECK: combiner {
|
|
// CHECK: ^bb0(%[[ARG0:.*]]: vector<2xi1>, %[[ARG1:.*]]: vector<2xi1>):
|
|
// CHECK: %[[RES:.*]] = arith.andi %[[ARG0]], %[[ARG1]] : vector<2xi1>
|
|
// CHECK: omp.yield(%[[RES]] : vector<2xi1>)
|
|
// CHECK: }
|
|
// CHECK-NOT: atomic
|
|
|
|
func.func @vector_and_reduction() {
|
|
%v_mask = vector.constant_mask [1] : vector<2xi1>
|
|
%c0 = arith.constant 0 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c2 = arith.constant 2 : index
|
|
%result = scf.parallel (%i) = (%c0) to (%c2) step (%c1) init(%v_mask) -> vector<2xi1> {
|
|
%val = vector.constant_mask [1] : vector<2xi1>
|
|
scf.reduce (%val : vector<2xi1>) {
|
|
^bb0(%lhs: vector<2xi1>, %rhs: vector<2xi1>):
|
|
%0 = arith.andi %lhs, %rhs : vector<2xi1>
|
|
scf.reduce.return %0 : vector<2xi1>
|
|
}
|
|
}
|
|
return
|
|
} |