xref: /llvm-project/mlir/test/Conversion/ConvertToSPIRV/simple.mlir (revision 25ae1a266d50f24a8fffc57152d7f3c3fcb65517)
1// RUN: mlir-opt -test-convert-to-spirv="run-signature-conversion=false run-vector-unrolling=false" -split-input-file %s | FileCheck %s
2
3// CHECK-LABEL: @return_scalar
4// CHECK-SAME: %[[ARG0:.*]]: i32
5// CHECK: spirv.ReturnValue %[[ARG0]]
6func.func @return_scalar(%arg0 : i32) -> i32 {
7  return %arg0 : i32
8}
9
10// CHECK-LABEL: @return_vector
11// CHECK-SAME: %[[ARG0:.*]]: vector<4xi32>
12// CHECK: spirv.ReturnValue %[[ARG0]]
13func.func @return_vector(%arg0 : vector<4xi32>) -> vector<4xi32> {
14  return %arg0 : vector<4xi32>
15}
16